Skip to content

Commit 20bf12d

Browse files
authored
Merge pull request Cadene#57 from tfriedel/master
bugfix: nasnet mobile was not listed in pretrainedmodels.model_names
2 parents 25f1466 + 9d4ea5b commit 20bf12d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pretrainedmodels/models/utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from .inceptionv4 import pretrained_settings as inceptionv4_settings
55
from .inceptionresnetv2 import pretrained_settings as inceptionresnetv2_settings
66
from .torchvision_models import pretrained_settings as torchvision_models_settings
7-
from .nasnet_mobile import pretrained_settings as nasnet_settings
7+
from .nasnet_mobile import pretrained_settings as nasnet_mobile_settings
88
from .nasnet import pretrained_settings as nasnet_settings
99
from .dpn import pretrained_settings as dpn_settings
1010
from .xception import pretrained_settings as xception_settings
@@ -17,6 +17,7 @@
1717
inceptionv4_settings,
1818
inceptionresnetv2_settings,
1919
torchvision_models_settings,
20+
nasnet_mobile_settings,
2021
nasnet_settings,
2122
dpn_settings,
2223
xception_settings,

0 commit comments

Comments
 (0)