Anaconda中修改PyTorch下载预训练模型的默认保存路径

最新推荐文章于 2024-03-17 11:56:17 发布
原创 最新推荐文章于 2024-03-17 11:56:17 发布 · 2.6k 阅读 · 12 ·
本内容遵循CC 4.0 BY-SA版权协议
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
标签
#pytorch #深度学习 #机器学习
本文介绍如何修改PyTorch预训练模型的默认下载路径，避免占用C盘空间。通过修改load_state_dict_from_url函数中的model_dir参数，可以指定模型下载到任意目录。

今天下载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 = 想要的模型下载绝对路径即可。

Resnet 官方代码 pytorch
苏凯祺的博客
10-26 7231
import torch import torch.nn as nn #from .utils import load_state_dict_from_url #其中torch.nn 为其提供基础函数，model_zoo提供权重数据的下载。 __all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101', ...
AlexNet原理及Pytorch实现
群星闪耀的博客
05-12 688
1.论文分析 为了对卷积神经网络CNN有更加透彻的了解，方便后续的CNN应用在其他方面。决定对经典论文AlexNet重新实现一遍。论文中原图如下： The first convolutional layer filters the 224×224×3 input image with 96 kernels of size 11×11×3 with a stride of 4 pixels (this is the distance between the receptive field cente
Pytorch预训练模型(torch.hub)缓存地址修改
JustPeanut的博客
04-15 8558
介绍了Pytorch是如何设置默认缓存路径的。
Pytorch实现动物识别(含动物数据集和训练代码)
热门推荐
AI吃大瓜的博客
09-05 2万+
动物识别数据集，动物分类数据集，动物数据集，动物分类识别，animal dataset,动物识别数据集+动物分类识别训练代码 Pytorch 支持 googlenet, resnet, inception_v3, mobilenet_v2
VGG代码详解
Bungehurst
03-12 6077
Pytorch官方实现 首先由引入相关的库 import torch import torch.nn as nn from .utils import load_state_dict_from_url 这里用到
【Pytorch】加载torchvision中预训练好的模型并修改默认下载路径（使用models.__dict__[model_name]()读取）
ProLover98的博客
03-11 1万+
说明 使用torchvision.model加载预训练好的模型时，发现默认下载路径在系统盘下面的用户目录下（这个你执行的时候就会发现），即C:\用户名\.cache\torch\.checkpoints下，便于统一管理，我决定修改model的存放路径，在网上找了很久都没有很好的解决方法，只能自己尝试，现将解决方案给出，供大家参考~ 操作环境 windows10 + Anaconda torch:...
基于Anaconda创建的虚拟环境，修改预训练模型下载的路径
weixin_44984048的博客
01-23 1798
pytorch修改预训练模型的保存路径
pytorch 安装 Bert安装 BERT Pytorch版本手动安装bert-base-uncased模型 总结
fghjbjhgb的博客
08-06 1万+
PyTorch环境配置及安装 点这，安装图文. 大概过程 1.首先使用 conda 指令创建一个屋子：conda create -n pytorch python=3.6 2.进入屋内：conda activate pytorch 3.安装pytorch 4.cuda选择10.1 上面链接到第六步加速的时候看下面的链接 看这，点击 从补充说明开始看 按这个进行安装，安装失败，pytorch下到一半就下不动了。到镜像地址去下载镜像，点这， 复制放到到Anaconda 安装的目录下的 pkgs 文件夹中。选择替
武大超算linux系统平台配置pytorch环境
Cyj_0703的博客
10-11 1340
武大超算linux系统平台配置pytorch环境
ResNet代码解析（Pytorch）
学习 & 分享 ~
09-06 1万+
源代码：models/resnet.pytorchvision 是 pytorch 中一个很好用的包，主要由 3 个子包，分别是 ， 和 参考官网：https://pytorch.org/vision/stable/index.html 代码：https://github.com/pytorch/vision/tree/master/torchvision在 torchvision 中实现了几个模型，包含 AlexNet，DenseNet，ResNet，VGG 等常用结构，并提供了预训练模型。 导入模型：
No module named ‘torchvision.models.utils‘,无法加载load_state_dict_from_url解决
weixin_49252254的博客
02-16 6185
No module named ‘torchvision.models.utils’,无法加载load_state_dict_from_url解决 由于torchvision的版本更新，对应的函数转移到了新版本的torch里面 from torchvision.models.utils import load_state_dict_from_url 改 from torch.hub import load_state_dict_from_url ...
修改pycharm下载预训练模型的保存路径
weixin_62769552的博客
04-21 3120
通过对源码进行修改来改变pycharm下载预训练模型的保存地址
【PyTorch】进阶学习：一文详细介绍 load_state_dict() 的应用场景、实战代码示例
创作高质量博文，分享知识，共同进步！
03-17 5403
🚀【PyTorch进阶】一文解锁load_state_dict()应用奥秘🔓 深度学习高手必备技能！一文带你探索PyTorch中load_state_dict()的实战应用场景！从模型迁移学习到微调，再到多模型集成与参数共享，让你轻松驾驭模型参数的加载与迁移。💪 文章还包含丰富的实战代码示例，让你边学边做，快速上手！同时，注意事项与常见问题一网打尽，助你避开坑点，提升训练效率。🔍 进阶技巧与扩展应用更是让你大开眼界，开启深度学习新篇章！快来学习，成为PyTorch高手吧！#PyTorch #l
from torchvision.models.utils import load_state_dict_from_url 报错ModuleNotFoundError
啊呜兮兮的博客
04-14 2159
极简 from torchvision.models.utils import load_state_dict_from_url 报错ModuleNotFoundError:No module named 'torchvison.models.utils >> torchvision.__version__ >>'0.11.3 版本不支持 我降低到1.8.0就ok了
pytorch load state dict_从零开始学Pytorch(十六）
weixin_39616045的博客
11-21 617
优化器01概念 机器学习模块中，数据读取，构建模型，得到损失函数后，就需要构造优化器，管理并更新模型中可学习参数的值，使得模型输出更接近真实标签。pytorch的优化器:管理并更新模型中可学习参数的值，使得模型输出更接近真实标签。深度学习优化器的两个核心是梯度与学习率，前者决定参数更新的方向，后者决定参数更新程度。采用梯度原因：对于高维的函数其更高阶导的计算复杂度大，应用到深度学习的优...
神经网络load_state_dict()进阶使用
weixin_45007507的博客
09-02 1万+
神经网络load_state_dict()进阶使用
修改PyTorch预训练模型默认下载位置）
qq_2992427261
10-10 4560
说明 使用torchvision.model加载预训练好的模型时，发现默认下载路径在系统盘下面的用户目录下（这个你执行的时候就会发现），即模型下载的默认路径 C:\用户名.cache\torch.checkpoints下， 修改方法 发现pytorch的默认下载路径是由load_state_dict_from_url函数进行控制，那么就好办了，只需要找到这个函数进行修改即可 由于我是下载vgg16，所以我先找到vgg.py源码，位于python路径 torchvision/models/vgg.py(
deep learning coursera yolo.h5已训练模型的的导入问题
overfit的博客
12-07 7167
本人小白，使用的是windows10，anaconda3，在deep learning coursera作业终于到了h5文件的导入问题： 1.激活环境（如下图有小括号显示） 如果没有小括号显示可以进到Anaconda navigator 中的Environments，启动任意一个环境： 2.根据https://github.com/allanzelener/YAD2K上的
pytorch预训练模型下载保存路径和路径更改
q4326rgeqaadfd的博客
02-18 9970
pytorch 预训练模型下载
Pytorch中更改预训练权重文件的下载位置
孟孟单单的博客
05-16 8100
目录1. 参考链接2. 更改方法3. 一个小技巧 1. 参考链接 Pytorch更改预训练权重下载位置 pytorch———修改预训练模型下载路径 2. 更改方法 在线加载的预训练权重默认存放位置在：C:\Users\[用户名]\.cache\torch\hub\checkpoints 方法1：在某个.py文件最上方加上以下代码 import os # 更改预训练下载位置 os.environ['TORCH_HOME'] = '/data/Your_User_Name/pretrained_weig
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值