torchvision自动下载预训练模型后的保存路径—urllib.error.URLError: ＜urlopen error [Errno -2] Name or service not know

问题描述

集群中计算节点无法联网，或者在某些网络不太好的情况下使用torchvision中预训练模型加载失败，这时候就需要我们离线下载模型pth文件后再迁移到服务器中正确的路径。

解决方法

这里以maskrcnn预训练模型为例，其他同理。
1.获取预训练模型网址并下载：
Anacondas/anaconda3/envs/envsname/lib/python3.7/site-packages/torchvision/models/detection/mask_rcnn.py
代码中搜索关键字model_urls，这个字典中就含有下载网址。
2.迁移模型pth文件到服务器中的正确路径
默认路径： 用户名/.cache/torch/hub/checkpoints
(The default dir is <hub_dir>/checkpoints )