fixed minor mobilenet issues

wodaka · Sep 2, 2019 · 3448e8a · 3448e8a
1 parent d76d9c6
commit 3448e8a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/models/networks.py b/models/networks.py
@@ -276,8 +276,7 @@ def get_generator(model_config):
                                   n_blocks=model_config['blocks'],
                                   learn_residual=model_config['learn_residual'])
     elif generator_name == 'fpn_mobilenet':
-        model_g = FPNMobileNet(norm_layer=get_norm_layer(norm_type=model_config['norm_layer']),
-                               pretrained=model_config['pretrained'])
+        model_g = FPNMobileNet(norm_layer=get_norm_layer(norm_type=model_config['norm_layer']))
     elif generator_name == 'fpn_inception':
         model_g = FPNInception(norm_layer=get_norm_layer(norm_type=model_config['norm_layer']))
     elif generator_name == 'fpn_inception_simple':