add pretrain_model url for 0.75 and 1.3

Oneflow-Inc · Nov 2, 2023 · df4df4d · df4df4d
1 parent af65737
commit df4df4d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flowvision/models/mnasnet.py b/flowvision/models/mnasnet.py
@@ -14,9 +14,9 @@
 
 model_urls = {
     "mnasnet0_5": "https://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/MNASNet/mnasnet0_5.zip",
-    "mnasnet0_75": None,
+    "mnasnet0_75": "http://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/MNASNet/mnasnet0_75.zip",
     "mnasnet1_0": "https://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/MNASNet/mnasnet1_0.zip",
-    "mnasnet1_3": None,
+    "mnasnet1_3": "http://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/MNASNet/mnasnet1_3.zip",
 }
 
 _BN_MOMENTUM = 1 - 0.9997