@@ -103,7 +103,8 @@ def _cfg(url='', **kwargs):
103103 url = 'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b5-c6949ce9.pth' ,
104104 input_size = (3 , 456 , 456 ), pool_size = (15 , 15 ), crop_pct = 0.934 ),
105105 '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' ),
107108 'mixnet_l' : _cfg (url = '' ),
108109 'tf_mixnet_s' : _cfg (
109110 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):
271272 return [deepcopy (block_args ) for _ in range (num_repeat )]
272273
273274
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-
281275def _decode_arch_def (arch_def , depth_multiplier = 1.0 ):
282276 arch_args = []
283277 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):
16121606 model = _gen_mixnet_m (
16131607 channel_multiplier = 1.0 , num_classes = num_classes , in_chans = in_chans , ** kwargs )
16141608 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 )
16171611 return model
16181612
16191613
0 commit comments