[PyTorch]论文pytorch复现中遇到的BUG

最新推荐文章于 2023-02-18 14:50:08 发布
最新推荐文章于 2023-02-18 14:50:08 发布
阅读量2.2k 收藏 1
点赞数 1
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_44538273/article/details/88403951
版权
本文档记录了在PyTorch中复现论文实验时遇到的三个主要问题：1) 使用zip迭代错误，原因在于返回了常量而非可迭代对象；2) DataParallel使用不当，需先将模型移动到GPU后再并行化；3) 从单GPU模型state_dict加载到多GPU模型时，由于module前缀导致参数未更新，解决方案是编写通用的权重加载函数。
摘要由CSDN通过智能技术生成
最低0.47元/天 解锁文章
急流勇进
关注
【图像超分】论文复现：新手入门必备！Pytorch实现SRGAN！手把手详细教程！代码注释详尽！可用于训练自己的数据集！实时验证模型性能！绘制PSNR曲线图！附训练好的各放大倍数的模型权重文件！
畅游计算机视觉的海洋
04-15 4620
【图像超分】论文精读：SRResNet/SRGAN【图像超分】论文复现：万字长文！Pytorch实现SRResNet代码修改无报错！踩坑全记录！适合各种深度学习新手！帮助你少走弯路！附修改后的代码和PSNR最优的模型权重文件！请配合上述论文精读文章使用，效果更佳！与原论文描述不一样的实现部分，也即可以修改的部分数据集使用：论文中是ImageNet随机选35000张图像，本文是VOC2012数据集16700张图像。数据集预处理：裁剪大小不同，论文是96×96，本文是88×88。
pytorch模型加载测试_pytorch模型加载方法汇总
weixin_39914975的博客
12-19 1375
Pytorch有很多方便易用的包，今天要谈的是torchvision包，它包括3个子包，分别是： torchvison.datasets ，torchvision.models ，torchvision.transforms ，分别是预定义好的数据集(比如MNIST、CIFAR10等)、预定义好的经典网络结构(比如AlexNet、VGG、ResNet等)和预定义好的数据增强方法(比如Resize、...
运行深度学习程序，最后输入一张图片进行结果验证，这是什么错误
zhangyue880820的博客
12-04 1316
Traceback (most recent call last): File "/home/zhangyue/shiyan1/predict.py", line 67, in <module> r_image = frcnn.detect_image(image) File "/home/zhangyue/shiyan1/frcnn.py", line 143, in detect_image roi_cls_locs, roi_scores, rois, _ = se...
TypeError: zip argument #1 must support iteration
SoP_KaiXinHaHa的博客
03-19 2447
跑M3D-RPN遇到了一个奇怪的bug，之前运行都好好的，忽然重新运行时报错， [Previous line repeated 1 more time] TypeError: zip argument #1 must support iteration 原因是没加CUDA_VISIBLE_DEVICES=0 ...
解决：zip argument #1 must support iteration
菜鸟IT女
11-08 1万+
问题 今儿做数据可视化，代码感觉正常的不得了。结果报错如下： 哽咽~就寻思着，为什么第一个for循环的zip函数没有任何问题，第二个for循环的zip函数就有问题了呢？ 解决思路 看报错信息，是说第二个zip里面有元素是不可迭代的，不过也看不出啥…结果定位到第一个for循环的变量名为x，刚好与变量x重名… 于是我悟了，for循环的本质是利用__next__，所以循环完毕之后到最后一个__next__，也就是30，也就是说，x=30 这本来一点问题都木有，直到下面紧接着又往zip里面传x，此时就有问题了。因
pytorch 确保结果可复现笔记
不解不惑
02-18 790
在使用pytorch进行深度神经网络学习时，遇到了代码不能复现的情况，很是令人烦恼，经过一番实验，找到了一种设置方法。
图像超分辨率算法EDSR的Pytorch版本复现代码，适合新手，修改后无bug，含制作好的h5数据集和最优PSNR的模型权重文件
04-12
pytorch环境：torch1.9.1+cuda11.1 再安装好所需的包即可运行。 文件目录： data：数据集和生成h5数据集的matlab文件 Set5：测试集Set5（matlab生成的.mat格式，不用matlab生成这个格式计算的PSNR有偏差，这样最准...
图像超分辨率SRResNet的Pytorch版本复现代码，适合新手入门，修改后无bug，含最优PSNR的模型权重文件
最新发布
04-12
pytorch环境：torch1.9.1+cuda11.1 再安装好所需的包即可运行。 文件目录： checkpoint：存放训练过程中的模型文件 data：h5格式训练集位置，生成h5格式训练集的matlab代码 model：作者给出的模型文件位置 testsets...
图像超分辨率算法LapSRN的Pytorch版本复现代码，适合新手入门，修改后无bug，含最优PSNR的模型权重文件
04-12
pytorch环境：torch1.9.1+cuda11.1 再安装好所需的包即可运行。 文件目录： checkpoint：存放训练过程中的模型文件 data：h5格式训练集位置，生成h5格式训练集的matlab代码 model：作者给出的模型文件位置 testsets...
pytorch中的pre-train函数模型引用及修改（增减网络层，修改某层参数等）
热门推荐
whut_ldz的博客
12-19 7万+
一、pytorch中的pre-train模型 卷积神经网络的训练是耗时的，很多场合不可能每次都从随机初始化参数开始训练网络。 pytorch中自带几种常用的深度学习网络预训练模型，如VGG、ResNet等。往往为了加快学习的进度，在训练的初期我们直接加载pre-train模型中预先训练好的参数，model的加载如下所示： import torchvision.models as mo
Python 基础学习 内置函数总结笔记
One Way的博客
02-04 755
常用函数 zip() zip()函数将可迭代的对象对应打包 a = ['a','b','c'] b = [1,2,3,4,5] c = zip(a,b) print(c) print(list(c)) e = 5 f = 6 print(list(zip(e,f))) outcome: &amp;amp;lt;zip object at 0x7f0332182708&amp;amp;gt; [('a', 1), ('b', 2...
编码过程中的一些问题解决方法记录
TTTree_的博客
10-18 2977
1. UnicodeDecodeError: ‘gbk’ codec can’t decode byte 0x99 in position 10856: illegal multibyte sequence 在安装sacred时出错，依照官网教程，pip install sacred，出现此错误。 解决方法：找到报错所在的.py代码，一般是open文件的地方出错，将open的打开编码方式改为：en...
pytorch模型加载测试_记录一次pytorch加载模型的采坑
weixin_39691968的博客
12-19 880
因为我有多块gpu，想都用来做训练，查阅资料，得用到dataparallel这个方法，刚好从github上找到了一个比较完整的pytorch图像分类代码也是使用的dataparallel方法，就直接拿来试验。当训练完毕储存模型，想再次加载测试一下结果的时候，evaluate的结果基本都是1%的情况，这让我很费解，训练完毕的时候可是有73%呢，又是就开始了长达整个半天的采坑之旅。原始加载模型代码如下...
PyTorch: param_groups = list(params) TypeError: 'method' object is not iterable
RaySunWHUT
02-24 3957
Question: ```handlebars Traceback (most recent call last): File "G:/pycharm Project/PytorchLearning/first/Classifier.py", line 210, in <module> main() File "G:/pycharm Project/PytorchLea...
pytorch使用过程中遇到的警告
legalhighhigh的博客
11-04 1万+
1、UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number #原语句： train_loss+=loss.data[0] #修改后： train_lo...
pytorch yolov5代码复现
06-06
pytorch yolov5代碼是一個基於pytorch框架的實現YOLOv5的代碼庫，它可以用於目標檢測任務，比如在圖片中識別出物體的位置、大小、類型等信息。該代碼庫的重現了論文《YOLOv5: An Incremental Improvement》中提出的...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值