From a0275cfa2fff5c7b4c3c179641eb2d16ec457fa4 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 24 Jun 2019 08:25:14 -0700 Subject: [PATCH] Fix arg positions in two entrypoint aliases --- timm/models/gen_efficientnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/gen_efficientnet.py b/timm/models/gen_efficientnet.py index 12d280fc..05a7c2ce 100644 --- a/timm/models/gen_efficientnet.py +++ b/timm/models/gen_efficientnet.py @@ -1189,7 +1189,7 @@ def mnasnet_100(pretrained=False, num_classes=1000, in_chans=3, **kwargs): @register_model def mnasnet_b1(pretrained=False, num_classes=1000, in_chans=3, **kwargs): """ MNASNet B1, depth multiplier of 1.0. """ - return mnasnet_100(num_classes, in_chans, pretrained, **kwargs) + return mnasnet_100(pretrained, num_classes, in_chans, **kwargs) @register_model @@ -1253,7 +1253,7 @@ def semnasnet_100(pretrained=False, num_classes=1000, in_chans=3, **kwargs): @register_model def mnasnet_a1(pretrained=False, num_classes=1000, in_chans=3, **kwargs): """ MNASNet A1 (w/ SE), depth multiplier of 1.0. """ - return semnasnet_100(num_classes, in_chans, pretrained, **kwargs) + return semnasnet_100(pretrained, num_classes, in_chans, **kwargs) @register_model