From df4df4dfff2a15cf6bae96e81e7a1bab548cd207 Mon Sep 17 00:00:00 2001 From: cuirunxi <953131037@qq.com> Date: Thu, 2 Nov 2023 06:41:43 +0000 Subject: [PATCH] add pretrain_model url for 0.75 and 1.3 --- flowvision/models/mnasnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flowvision/models/mnasnet.py b/flowvision/models/mnasnet.py index 8159b8b8..259df44d 100644 --- 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