今天下载vgg的模型参数时,发现默认下载到C盘中,然而家境贫寒C盘只有60G,前期不懂,什么都是默认地址,C盘只剩可怜的10G了,卡的要死。。。。
记录一下修改的过程,找了各种博文整了一小时。
1.在pycharm里找到venv\lib\torch\hub.py
2.在hub.py找到model_dir这里表示模型下载地址,为NONE的话表示默认下载到C盘cache中,可修改为自己想要下载的地址。
记住:要为单引号,且为/不然会报错(直接复制的路径为\)
最后要加checkpoints