@@ -103,7 +103,8 @@ def _cfg(url='', **kwargs):
103
103
url = 'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b5-c6949ce9.pth' ,
104
104
input_size = (3 , 456 , 456 ), pool_size = (15 , 15 ), crop_pct = 0.934 ),
105
105
'mixnet_s' : _cfg (url = '' ),
106
- 'mixnet_m' : _cfg (url = '' ),
106
+ 'mixnet_m' : _cfg (
107
+ url = 'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mixnet_m-4647fc68.pth' ),
107
108
'mixnet_l' : _cfg (url = '' ),
108
109
'tf_mixnet_s' : _cfg (
109
110
url = 'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_mixnet_s-89d3354b.pth' ),
@@ -271,13 +272,6 @@ def _decode_block_str(block_str, depth_multiplier=1.0):
271
272
return [deepcopy (block_args ) for _ in range (num_repeat )]
272
273
273
274
274
- def _decode_arch_args (string_list ):
275
- block_args = []
276
- for block_str in string_list :
277
- block_args .append (_decode_block_str (block_str ))
278
- return block_args
279
-
280
-
281
275
def _decode_arch_def (arch_def , depth_multiplier = 1.0 ):
282
276
arch_args = []
283
277
for stack_idx , block_strings in enumerate (arch_def ):
@@ -1612,8 +1606,8 @@ def mixnet_m(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
1612
1606
model = _gen_mixnet_m (
1613
1607
channel_multiplier = 1.0 , num_classes = num_classes , in_chans = in_chans , ** kwargs )
1614
1608
model .default_cfg = default_cfg
1615
- # if pretrained:
1616
- # load_pretrained(model, default_cfg, num_classes, in_chans)
1609
+ if pretrained :
1610
+ load_pretrained (model , default_cfg , num_classes , in_chans )
1617
1611
return model
1618
1612
1619
1613
0 commit comments