解决torch.hub.load加载网络模型异常

1 torch.hub.load 加载网络模型错误

通过网络使用torch.hub.load加载模型代码如下：

self.model = torch.hub.load("facebookresearch/dinov2", 'dinov2_vits14', source='github').to(self.device)

运行网上的项目，经常会卡住或者超时，原因是 torch.hub.load 默认会去网上找模型，而github经常是不可访问的（需要走代理），从而导致网络异常，错误如下：

Traceback (most recent call last):File "/opt/pa_retrieve/preprocessor/remove_redundant_image.py", line 4, in <module>from model.dinov2_embeding_small import dinov2_embeding_smallFile "/opt/pa_retrieve/model/dinov2_embeding_small.py", line 42, in <module>dinov2_embeding_small = Dinov2EmbedingSmall()File "/opt/pa_retrieve/model/dinov2_embeding_small.py", line 20, in __init__self.model = torch.hub.load("facebookresearch/dinov2", 'dinov2_vits14', source='github').to(self.device)File "/root/anaconda3/envs/pa/lib/python3.9/site-packages/torch/hub.py", line 555, in loadrepo_or_dir = _get_cache_or_reload(repo_or_dir, force_reload, trust_repo, "load",File "/root/anaconda3/envs/pa/lib/python3.9/site-packages/torch/hub.py", line 199, in _get_cache_or_reloadrepo_owner, repo_name, ref = _parse_repo_info(github)File "/root/anaconda3/envs/pa/lib/python3.9/site-packages/torch/hub.py", line 142, in _parse_repo_infowith urlopen(f"https://github.com/{repo_owner}/{repo_name}/tree/main/"):File "/root/anaconda3/envs/pa/lib/python3.9/urllib/request.py", line 214, in urlopenreturn opener.open(url, data, timeout)File "/root/anaconda3/envs/pa/lib/python3.9/urllib/request.py", line 517, in openresponse = self._open(req, data)File "/root/anaconda3/envs/pa/lib/python3.9/urllib/request.py", line 534, in _openresult = self._call_chain(self.handle_open, protocol, protocol +File "/root/anaconda3/envs/pa/lib/python3.9/urllib/request.py", line 494, in _call_chainresult = func(*args)File "/root/anaconda3/envs/pa/lib/python3.9/urllib/request.py", line 1389, in https_openreturn self.do_open(http.client.HTTPSConnection, req,File "/root/anaconda3/envs/pa/lib/python3.9/urllib/request.py", line 1350, in do_openr = h.getresponse()File "/root/anaconda3/envs/pa/lib/python3.9/http/client.py", line 1377, in getresponseresponse.begin()File "/root/anaconda3/envs/pa/lib/python3.9/http/client.py", line 320, in beginversion, status, reason = self._read_status()File "/root/anaconda3/envs/pa/lib/python3.9/http/client.py", line 289, in _read_statusraise RemoteDisconnected("Remote end closed connection without"
http.client.RemoteDisconnected: Remote end closed connection without response

2 torch.hub.load加载本地模型

通过代理下载模型和代码，模型存放在如下目录下：

/root/.cache/torch/hub/checkpoints/

工程代码存放在如下目录下：

/root/.cache/torch/hub/facebookresearch_dinov2_main/

更改模型加载的代码为本地加载，代码如下：

self.model = torch.hub.load('/root/.cache/torch/hub/facebookresearch_dinov2_main', 'dinov2_vits14', trust_repo=True, source='local').to(self.device)

再次运行程序，模型加载成功。