Anaconda中修改PyTorch下载预训练模型的默认保存路径

今天下载vgg16预训练模型时发现默认下载位置在C盘,这对C盘空间小的同学很不友好。

在这里插入图片描述下面介绍修改预训练模型默认保存路径的方法(其他模型同样操作)。PyTorch的默认下载路径由load_state_dict_from_url()函数确定,因此找到该函数进行修改即可。

在这里插入图片描述
按Ctrl+鼠标左键进入vgg.py源码,搜索load_state_dict_from_url,进入utils.py

在这里插入图片描述

进入hub.py

Anaconda中修改PyTorch下载预训练模型的默认保存路径_第1张图片

搜索load_state_dict_from_url

Anaconda中修改PyTorch下载预训练模型的默认保存路径_第2张图片

将model_dir = None换成model_dir = 想要的模型下载绝对路径即可。

你可能感兴趣的:(PyTorch,pytorch,深度学习,机器学习)