From f8b41a452c43809aafb1530b03e83122bba5ba35 Mon Sep 17 00:00:00 2001 From: Minghao Zhu Date: Sun, 12 Dec 2021 21:26:44 -0500 Subject: [PATCH 1/2] fix No module named 'torchvision.models.utils' error --- lanedet/models/backbones/mobilenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lanedet/models/backbones/mobilenet.py b/lanedet/models/backbones/mobilenet.py index f381ac8..f94b3d3 100644 --- a/lanedet/models/backbones/mobilenet.py +++ b/lanedet/models/backbones/mobilenet.py @@ -1,7 +1,7 @@ import torch from torch import nn from torch import Tensor -from torchvision.models.utils import load_state_dict_from_url +from torch.hub import load_state_dict_from_url from typing import Callable, Any, Optional, List from lanedet.models.registry import BACKBONES From 3ec77ead0242425151c10e9b19ba7057a432f477 Mon Sep 17 00:00:00 2001 From: Minghao Zhu <25329580+pgchui@users.noreply.github.com> Date: Mon, 21 Mar 2022 00:06:19 -0400 Subject: [PATCH 2/2] Update requirements.txt --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index b1f6ab2..cd3ca02 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ -torch==1.6.0 -torchvision==0.7.0 +torch==1.9.0 +torchvision==0.10.0 pandas addict sklearn