Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks.

pull/1606/head
Ross Wightman 2 years ago
parent c8f69e04a9
commit 4042a94f8f

@ -97,9 +97,11 @@ default_cfgs = {
url='', interpolation='bicubic'), url='', interpolation='bicubic'),
'cs3edgenet_x': _cfg( 'cs3edgenet_x': _cfg(
url='', interpolation='bicubic'), url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3edgenet_x_c2-2e1610a9.pth',
interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=1.0),
'cs3se_edgenet_x': _cfg( 'cs3se_edgenet_x': _cfg(
url='', interpolation='bicubic'), url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3se_edgenet_x_c2ns-76f8e3ac.pth',
interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0),
} }

Loading…
Cancel
Save