![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch学习笔记
文章平均质量分 53
揽风入怀
从现在起每周更新一篇笔记,好多记在云盘里了还不如放在CSDN方便查阅
展开
-
Pytorch将模型加载到GPU中训练时遇到的坑
1. 第一个容易遇到的坑: 你的target标签字典中含有list等非tensor的数据类型,这时候不处理会报错如下:(1)'dict' object has no attribute 'cuda'(2)'list' object has no attribute 'cuda'解决方法:(1)https://blog.csdn.net/york1996/article/det...原创 2020-03-25 19:12:03 · 14093 阅读 · 12 评论 -
Pytorch修改预训练模型的方法汇总
本文包括如何修改预训练模型的示例。常见的有四种不同程度的修改:1、只修改输入输出的类别数,即某些网络层的参数(常见的是修改通道数)2、替换整个backbone或预训练模型的某一部分3、修改网络中间层的结构(最重要,一般是重写部分中间层再整体替换)4、快速去除预训练模型本身的网络层并添加新的层正文如下1.只修改输入输出的类别数,即某些网络层的参数(常见的是修改通道数)...原创 2019-12-30 10:11:55 · 17314 阅读 · 7 评论 -
2019年Ubuntu 18安装Pytorch、torchvision的最快捷方法
1、Ubuntu系统及下载源优化(如果已经更换国内源,请忽略此步)#先备份原来的源列表文件sudo cp /etc/apt/sources.list /etc/apt/sourses.list.backup#新建源列表文件nano /etc/apt/sources.list#在该列表文件中加入源地址(推荐用阿里云源)deb http://mirrors.aliyun.com/ub...原创 2019-11-01 22:15:40 · 2426 阅读 · 1 评论 -
Pytorch保存、加载模型的方法
保存模型的推荐方法(转自Pytorch官方文档)这主要有两种方法序列化和恢复模型。第一种(推荐)只保存和加载模型参数:PATH = "/home/xx/results/param.pth"torch.save(the_model.state_dict(), PATH)#只保存模型参数然后,加载模型时使用:the_model = TheModelClass(*args, *...原创 2019-10-22 17:07:53 · 156 阅读 · 0 评论