Merge remote-tracking branch 'origin/fixes_bce_regnet' into bits_and_tpu

pull/1239/head
Ross Wightman 3 years ago
commit 52c481ea8e

@ -141,13 +141,16 @@ default_cfgs = {
# experimental models # experimental models
'regnetz_b': _cfg( 'regnetz_b': _cfg(
url='', url='',
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),
input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'), input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'),
'regnetz_c': _cfg( 'regnetz_c': _cfg(
url='', url='',
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),
input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'), input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'),
'regnetz_d': _cfg( 'regnetz_d': _cfg(
url='', url='',
input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'), mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),
first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'),
} }

Loading…
Cancel
Save