如何更改深度学习框架下载模型的默认位置(C盘)

我使用的是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虚拟环境,这个值就变成了:
在这里插入图片描述

改了之后重新执行下载模型的代码就会改变下载的地址:
在这里插入图片描述

  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
现代深度学习框架是构建复杂神经网络模型的关键工具,它们提供了一整套库和API,使得开发者能够方便地定义、训练和部署深度学习模型。以下是深度学习框架建立模型的一些关键步骤: 1. **选择框架**:常见的深度学习框架有TensorFlow、PyTorch、Keras(现在是TensorFlow的一个高级API)、Microsoft's ML.NET、Apache MXNet等。每种框架都有其优点和适用场景,选择取决于项目需求、社区支持和开发者的熟悉程度。 2. **模型架构设计**:使用框架提供的API,如PyTorch的nn.Module或TensorFlow的tf.keras.Model,设计神经网络的层次结构,包括输入层、隐藏层和输出层,可能还会包含卷积层、循环层(如LSTM或GRU)等。 3. **定义模型**:配置模型的参数,如层数、节点数、激活函数等,并可能使用预训练的权重进行迁移学习。 4. **数据准备**:整理和加载训练数据集,通常需要进行数据预处理,如标准化、归一化、数据增强等,以提高模型性能。 5. **编译模型**:为模型指定损失函数、优化器和评估指标,如交叉熵损失、Adam优化器、准确率等。 6. **训练模型**:通过训练数据对模型进行迭代训练,通常会设置验证集来监控模型在未见过的数据上的表现并防止过拟合。 7. **模型评估和调整**:在测试集上评估模型性能,根据结果调整模型参数或修改架构。 8. **模型保存与部署**:将训练好的模型保存到文件中,以便在其他环境或应用中使用,也可以通过封装为API或服务进行部署。 相关问题: 1. 深度学习框架有哪些常见的优点? 2. 如何在深度学习模型中实现自定义的层或模块? 3. 在训练过程中如何有效地防止过拟合?

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值