From a10edaf80408064cf53822c2370f9a0e113ee6b2 Mon Sep 17 00:00:00 2001 From: hetan697 <50706594+hetan697@users.noreply.github.com> Date: Tue, 7 Feb 2023 16:06:40 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=E4=B8=80=E5=A4=84=E8=BF=87?= =?UTF-8?q?=E6=97=B6=E7=9A=84torch=E8=AF=AD=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tllib/vision/models/segmentation/deeplabv2.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tllib/vision/models/segmentation/deeplabv2.py b/tllib/vision/models/segmentation/deeplabv2.py index 3e960688..2bd734b4 100644 --- a/tllib/vision/models/segmentation/deeplabv2.py +++ b/tllib/vision/models/segmentation/deeplabv2.py @@ -3,8 +3,7 @@ @contact: JiangJunguang1123@outlook.com """ import torch.nn as nn -from torchvision.models.utils import load_state_dict_from_url - +from torch.hub import load_state_dict_from_url model_urls = { 'deeplabv2_resnet101': 'https://cloud.tsinghua.edu.cn/f/2d9a7fc43ce34f76803a/?dl=1'