概述
在使用pytorch之时，免不了要加载预习训练好的默认模型，如使用models.squeezenet1_1，此时会默认将模型加载在~/.torch/models/文件夹下，可以使用对环境变量的操作将其路径改变，以加载到指定文件夹。
对于模型的默认路径，是通过环境变量TORCH_HOME进行操控。通过os控制命令行修改环境变量。
os.environ['TORCH_HOME'] = '/media/sunkj/WorkAndStudy/pytorch/models/'
再进行模型加载，即可。结果如下：
resnet34 = models.squeezenet1_1(pretrained=True, num_classes=1000)
Downloading: "https://download.pytorch.org/models/squeezenet1_1-f364aa15.pth" to /media/sunkj/WorkAndStudy/pytorch/models/checkpoints/squeezenet1_1-f364aa15.pth
100%|██████████| 4.74M/4.74M [00:01<00:00, 2.57MB/s]
最后
以上就是结实奇异果为你收集整理的pytorch加载默认模型位置的全部内容，希望文章能够帮你解决pytorch加载默认模型位置所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复