@@ -1441,12 +1441,6 @@ def mnasnet_100(pretrained=False, **kwargs) -> EfficientNet:
1441
1441
return model
1442
1442
1443
1443
1444
- @register_model
1445
- def mnasnet_b1 (pretrained = False , ** kwargs ) -> EfficientNet :
1446
- """ MNASNet B1, depth multiplier of 1.0. """
1447
- return mnasnet_100 (pretrained , ** kwargs )
1448
-
1449
-
1450
1444
@register_model
1451
1445
def mnasnet_140 (pretrained = False , ** kwargs ) -> EfficientNet :
1452
1446
""" MNASNet B1, depth multiplier of 1.4 """
@@ -1475,12 +1469,6 @@ def semnasnet_100(pretrained=False, **kwargs) -> EfficientNet:
1475
1469
return model
1476
1470
1477
1471
1478
- @register_model
1479
- def mnasnet_a1 (pretrained = False , ** kwargs ) -> EfficientNet :
1480
- """ MNASNet A1 (w/ SE), depth multiplier of 1.0. """
1481
- return semnasnet_100 (pretrained , ** kwargs )
1482
-
1483
-
1484
1472
@register_model
1485
1473
def semnasnet_140 (pretrained = False , ** kwargs ) -> EfficientNet :
1486
1474
""" MNASNet A1 (w/ SE), depth multiplier of 1.4. """
@@ -1590,13 +1578,6 @@ def efficientnet_b2(pretrained=False, **kwargs) -> EfficientNet:
1590
1578
return model
1591
1579
1592
1580
1593
- @register_model
1594
- def efficientnet_b2a (pretrained = False , ** kwargs ) -> EfficientNet :
1595
- """ EfficientNet-B2 @ 288x288 w/ 1.0 test crop"""
1596
- # WARN this model def is deprecated, different train/test res + test crop handled by default_cfg now
1597
- return efficientnet_b2 (pretrained = pretrained , ** kwargs )
1598
-
1599
-
1600
1581
@register_model
1601
1582
def efficientnet_b3 (pretrained = False , ** kwargs ) -> EfficientNet :
1602
1583
""" EfficientNet-B3 """
@@ -1606,13 +1587,6 @@ def efficientnet_b3(pretrained=False, **kwargs) -> EfficientNet:
1606
1587
return model
1607
1588
1608
1589
1609
- @register_model
1610
- def efficientnet_b3a (pretrained = False , ** kwargs ) -> EfficientNet :
1611
- """ EfficientNet-B3 @ 320x320 w/ 1.0 test crop-pct """
1612
- # WARN this model def is deprecated, different train/test res + test crop handled by default_cfg now
1613
- return efficientnet_b3 (pretrained = pretrained , ** kwargs )
1614
-
1615
-
1616
1590
@register_model
1617
1591
def efficientnet_b4 (pretrained = False , ** kwargs ) -> EfficientNet :
1618
1592
""" EfficientNet-B4 """
@@ -2361,4 +2335,8 @@ def tinynet_e(pretrained=False, **kwargs) -> EfficientNet:
2361
2335
'tf_efficientnetv2_m_in21k' : 'tf_efficientnetv2_m.in21k' ,
2362
2336
'tf_efficientnetv2_l_in21k' : 'tf_efficientnetv2_l.in21k' ,
2363
2337
'tf_efficientnetv2_xl_in21k' : 'tf_efficientnetv2_xl.in21k' ,
2338
+ 'efficientnet_b2a' : 'efficientnet_b2' ,
2339
+ 'efficientnet_b3a' : 'efficientnet_b3' ,
2340
+ 'mnasnet_a1' : 'semnasnet_100' ,
2341
+ 'mnasnet_b1' : 'mnasnet_100' ,
2364
2342
})
0 commit comments