From f3824fd353e3f58d13eeacc9a17e5741cb34a543 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=8B=E5=B9=B3=20=E6=8B=93=E5=AE=9F?= Date: Fri, 20 Dec 2019 12:02:06 +0900 Subject: [PATCH] fix mime-type bug Default inceptionv3() can't hand over num_classes to models.inception_v3() when pretrained == False So I added argument 'num_classes' when calling inception_v3() --- pretrainedmodels/models/torchvision_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pretrainedmodels/models/torchvision_models.py b/pretrainedmodels/models/torchvision_models.py index f9347161..f4d2178e 100644 --- a/pretrainedmodels/models/torchvision_models.py +++ b/pretrainedmodels/models/torchvision_models.py @@ -253,7 +253,7 @@ def inceptionv3(num_classes=1000, pretrained='imagenet'): r"""Inception v3 model architecture from `"Rethinking the Inception Architecture for Computer Vision" `_. """ - model = models.inception_v3(pretrained=False) + model = models.inception_v3(num_classes=num_classes,pretrained=False) if pretrained is not None: settings = pretrained_settings['inceptionv3'][pretrained] model = load_pretrained(model, num_classes, settings)