我使用的是python语言,使用的框架是MXNwet。
问题阐述
使用官方文档提供的下载预训练模型时,默认下载的位置为C:\\Users\\Admin\\AppData\\Roaming
:
但是我的C盘不够用,如何解决?
解决方式
在安装了mxnet之后(其他不同的框架可以参照类似的方式进行更改),找到安装路径:
D:\miniconda3\envs\mxnet-gpu\Lib\site-packages\mxnet
然后发现是mxnet.gluon.model_zoo中的vision导致下载的,所以安装的路径一定在这里可以找到:
这里用的是resnet,打开resnet.py查看,在这些框架中不会直接显示的写出来C:\\Users\\Admin\\AppData\\Roaming
,很可能都是用os
包来实现的,搜索os:
看来是base的data_dir函数的问题,搜索base:
在python中,一个.
表示当前文件所在的目录,两个表示回退一级,这里四个表示回退3级,然后可以发现base.py:
打开这个文件,找到data_dir函数:
发现调用了data_dir_default函数,找到这个函数:
这里用了os.environ.get,打印os.environ试试,可以看到environ是一个字典,所以这里传入"APPDATA",直接获取了C:\\Users\\Admin\\AppData\\Roaming
这个路径,现在更改这个参数,比如我改成:CONDA_PREFIX
,所对应的value为:
D:\\miniconda3\\envs\\mxnet-gpu
这个可能是不同的虚拟环境所对应的这个键的值都会不同,后来经过测试发现猜想是对的,比如我激活了web虚拟环境,这个值就变成了:
改了之后重新执行下载模型的代码就会改变下载的地址: