From 807647dcd1fb51f70c5ac7915dfccd0d6f8f8249 Mon Sep 17 00:00:00 2001
From: Guy Jacob <guy.jacob@intel.com>
Date: Wed, 3 Apr 2019 17:30:48 +0300
Subject: [PATCH] Bugfix in create_model()
Don't pass 'pretrained' parameter to our internal ImageNet models
(aka 'imagenet_extra_models') - they don't support it, and in any
case it's redundant as they're created only if pretrained == False
---
distiller/models/__init__.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/distiller/models/__init__.py b/distiller/models/__init__.py
index 3ad0ceb..30e2c8d 100755
--- a/distiller/models/__init__.py
+++ b/distiller/models/__init__.py
@@ -67,7 +67,7 @@ def create_model(pretrained, dataset, arch, parallel=True, device_ids=None):
elif arch in torch_models.__dict__:
model = torch_models.__dict__[arch](pretrained=pretrained)
elif (arch in imagenet_extra_models.__dict__) and not pretrained:
- model = imagenet_extra_models.__dict__[arch](pretrained=pretrained)
+ model = imagenet_extra_models.__dict__[arch]()
elif arch in pretrainedmodels.model_names:
model = pretrainedmodels.__dict__[arch](
num_classes=1000,
--
GitLab