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

原创

已于 2023-11-22 14:31:58 修改 · 1w 阅读

收录于

当前文章被以下社区和专栏收录：

于 2023-11-06 19:14:25 首次发布

文章讲述了在使用torch.hub.load加载PyTorch模型时遇到的问题，如网络连接不稳定导致的下载失败，以及如何通过本地加载解决。同时介绍了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_small
  File "/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 load
    repo_or_dir = _get_cache_or_reload(repo_or_dir, force_reload, trust_repo, "load

标签