【Pytorch】加载torchvision中预训练好的模型并修改默认下载路径(使用models.__dict__[model_name]()读取)

本文介绍了如何在Windows10环境下,使用Anaconda的Python环境中,修改PyTorch torchvision加载预训练模型时的默认下载路径。通过查找源码并修改相应函数,实现模型保存在指定目录下,以方便统一管理。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

说明

使用torchvision.model加载预训练好的模型时,发现默认下载路径在系统盘下面的用户目录下(这个你执行的时候就会发现),即C:\用户名\.cache\torch\.checkpoints下,便于统一管理,我决定修改model的存放路径,在网上找了很久都没有很好的解决方法,只能自己尝试,现将解决方案给出,供大家参考~

操作环境

  • windows10 + Anaconda
  • torch:1.1.0
  • torchvision:0.3.0

加载方式

以加载vgg16为例,首先定义网络结构

class CNN(nn.Module):

    def __init__(self, usegpu=
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值