From fb1027de4c13c745a9e5c0376742149f3c781dbe Mon Sep 17 00:00:00 2001 From: Cathesilta Date: Fri, 3 Dec 2021 10:30:52 +0800 Subject: [PATCH] Debugged a torchvision version problem (#600) Co-authored-by: wangshen --- fastreid/modeling/backbones/mobilenetv3.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/fastreid/modeling/backbones/mobilenetv3.py b/fastreid/modeling/backbones/mobilenetv3.py index 11e912a..5e659e7 100644 --- a/fastreid/modeling/backbones/mobilenetv3.py +++ b/fastreid/modeling/backbones/mobilenetv3.py @@ -4,7 +4,15 @@ from typing import Any, Callable, Dict, List, Optional, Sequence import torch from torch import nn, Tensor from torch.nn import functional as F -from torchvision.models.utils import load_state_dict_from_url + +#The style of importing Considers compatibility for the diversity of torchvision versions +try: + from torchvision.models.utils import load_state_dict_from_url +except ImportError: + try: + from torch.hub import load_state_dict_from_url + except ImportError: + from torch.utils.model_zoo import load_url as load_state_dict_from_url from fastreid.layers import get_norm from .build import BACKBONE_REGISTRY