add pretrained model urls for mobilenetv2.

chenyaofo · Apr 15, 2021 · edb09d1 · edb09d1
1 parent 96ee94d
commit edb09d1
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/pytorch_cifar_models/mobilenetv2.py b/pytorch_cifar_models/mobilenetv2.py
@@ -45,17 +45,17 @@
 
 
 cifar10_pretrained_weight_urls = {
-    'mobilenetv2_x0_5': '',
-    'mobilenetv2_x0_75': '',
-    'mobilenetv2_x1_0': '',
-    'mobilenetv2_x1_4': '',
+    'mobilenetv2_x0_5': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar10_mobilenetv2_x0_5-ca14ced9.pt',
+    'mobilenetv2_x0_75': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar10_mobilenetv2_x0_75-a53c314e.pt',
+    'mobilenetv2_x1_0': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar10_mobilenetv2_x1_0-fe6a5b48.pt',
+    'mobilenetv2_x1_4': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar10_mobilenetv2_x1_4-3bbbd6e2.pt',
 }
 
 cifar100_pretrained_weight_urls = {
-    'mobilenetv2_x0_5': '',
-    'mobilenetv2_x0_75': '',
-    'mobilenetv2_x1_0': '',
-    'mobilenetv2_x1_4': '',
+    'mobilenetv2_x0_5': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar100_mobilenetv2_x0_5-9f915757.pt',
+    'mobilenetv2_x0_75': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar100_mobilenetv2_x0_75-9ab3e178.pt',
+    'mobilenetv2_x1_0': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar100_mobilenetv2_x1_0-1311f9ff.pt',
+    'mobilenetv2_x1_4': 'https://github.com/chenyaofo/pytorch-cifar-models/releases/download/mobilenetv2/cifar100_mobilenetv2_x1_4-8a269f5e.pt',
 }