今天下载vgg16预训练模型时发现默认下载位置在C盘,这对C盘空间小的同学很不友好。
下面介绍修改预训练模型默认保存路径的方法(其他模型同样操作)。PyTorch的默认下载路径由load_state_dict_from_url()函数确定,因此找到该函数进行修改即可。
按Ctrl+鼠标左键进入vgg.py源码,搜索load_state_dict_from_url,进入utils.py
进入hub.py
搜索load_state_dict_from_url
将model_dir = None换成model_dir = 想要的模型下载绝对路径即可。



