windows下修改pytorch中torchvision预训练模型的存储路径

windows下pytorch使用torchvision.model加载官方预训练模型时,发现默认下载路径在系统盘下面的用户目录下(C:\Users\用名名/.cache\torch\checkpoints)。为了防止系统盘臃肿,现将此目录进行修改,操作如下:
1.使用Everything软件搜索hub.py,找到所使用对应python解释器的那个。
windows下修改pytorch中torchvision预训练模型的存储路径_第1张图片
2.对hub.py进行编辑,搜索load_state_dict_from_url函数,将其参数model_dir由None改为自己设定目录即可,保存退出即可。
windows下修改pytorch中torchvision预训练模型的存储路径_第2张图片

你可能感兴趣的:(深度学习,深度学习)