Add some more small model weights lcnet, mnas, mnv2

huggingface · Jan 15, 2022 · 9ca3437 · 9ca3437
1 parent fa6463c
commit 9ca3437
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py
@@ -73,7 +73,8 @@ def _cfg(url='', **kwargs):
     'mnasnet_140': _cfg(url=''),
 
     'semnasnet_050': _cfg(url=''),
-    'semnasnet_075': _cfg(url=''),
+    'semnasnet_075': _cfg(
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/semnasnet_075-18710866.pth'),
     'semnasnet_100': _cfg(
         url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth'),
     'semnasnet_140': _cfg(url=''),
@@ -83,7 +84,9 @@ def _cfg(url='', **kwargs):
     'mobilenetv2_035': _cfg(
         url=''),
     'mobilenetv2_050': _cfg(
-        url=''),
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv2_050-3d30d450.pth',
+        interpolation='bicubic',
+    ),
     'mobilenetv2_075': _cfg(
         url=''),
     'mobilenetv2_100': _cfg(

diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py
@@ -85,8 +85,14 @@ def _cfg(url='', **kwargs):
         input_size=(3, 240, 240), test_input_size=(3, 288, 288), crop_pct=0.95),
 
     "lcnet_035": _cfg(),
-    "lcnet_050": _cfg(),
-    "lcnet_075": _cfg(),
+    "lcnet_050": _cfg(
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_050-f447553b.pth',
+        interpolation='bicubic',
+    ),
+    "lcnet_075": _cfg(
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_075-318cad2c.pth',
+        interpolation='bicubic',
+    ),
     "lcnet_100": _cfg(
         url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_100-a929038c.pth',
         interpolation='bicubic',