mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix arg positions in two entrypoint aliases

Browse Source
This commit is contained in:
Ross Wightman 2019-06-24 08:25:14 -07:00
parent fe59249701
commit a0275cfa2f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/gen_efficientnet.py
View File

@ -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