[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模型加载测试_pytorch模型加载方法汇总
weixin_39914975的博客
12-19 1388
Pytorch有很多方便易用的包，今天要谈的是torchvision包，它包括3个子包，分别是： torchvison.datasets ，torchvision.models ，torchvision.transforms ，分别是预定义好的数据集(比如MNIST、CIFAR10等)、预定义好的经典网络结构(比如AlexNet、VGG、ResNet等)和预定义好的数据增强方法(比如Resize、...
运行深度学习程序，最后输入一张图片进行结果验证，这是什么错误
zhangyue880820的博客
12-04 1336
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 2472
跑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 818
在使用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...
YOLODet-PyTorch是端到端基于pytorch框架复现yolo最新算法的目标检测开发套件
最新发布
06-03
6. **社区支持与更新**：作为开源项目，YOLODet-PyTorch可能有一个活跃的开发者社区，不断更新和完善代码，修复bug，添加新特性，以跟上Yolo系列算法的最新进展。 7. **文件结构**："yolodet-pytorch-master"这一...
pytorch中的pre-train函数模型引用及修改（增减网络层，修改某层参数等）
热门推荐
whut_ldz的博客
12-19 7万+
一、pytorch中的pre-train模型 卷积神经网络的训练是耗时的，很多场合不可能每次都从随机初始化参数开始训练网络。 pytorch中自带几种常用的深度学习网络预训练模型，如VGG、ResNet等。往往为了加快学习的进度，在训练的初期我们直接加载pre-train模型中预先训练好的参数，model的加载如下所示： import torchvision.models as mo
TypeError: zip argument #N must support iteration
Lovemyse1f的博客
05-11 1723
观察: 正常的replaybuffer中每次都会存None,而添加的HER的额外样本后出现了这个问题. 打印batch发现存在很多的none样本. 原本buffer采用数组,后改为队列存储,当满的时候会将最先进去的样本数组踢出,故尝试取消None空样本.跑了20eps没有问题. 错误定位: state, action, reward, next_state, done = map(np.stack, zip(*batch)) #numpy.stack(arrays, axis=0)沿着新轴连接数组的序列。每
风格迁移 RuntimeError: storage has wrong size: expected 16825408 got 256
longshaonihaoa的博客
08-15 1万+
闲着试一下风格迁移，pytorch0.4，python3，windows系统 尝试使用了： http://www.pytorchtutorial.com/pytorch-style-transfer/ https://blog.csdn.net/zzlyw/article/details/78849250 提供的代码，在此先感谢两篇文章的作者 但是都遇到了 ...(之前省略) ...
GitHub学习中的一些错误及解决方法
longshaonihaoa的博客
02-25 1483
今天根据廖雪峰老师的教程https://www.liaoxuefeng.com/wiki/0013739516305929606dd18361248578c67b8067c8c017b000学习了一些GitHub的知识，先将学习过程中的一些问题及解决进行分享1、fatal: Not a git repository (or any of the parent directories): .git ...
RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cpu and
just_h的博客
03-28 1万+
RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cpu and parameter tensor at cuda:0 在学习pytorch的时候遇到的错误，意思是，输入和参数的张量不是在同一个device里，有一部分在CPU有一部分在GPU. 下面是一个例子，参考...
pytorch 踩坑笔记
Huang_Fj的博客
06-22 2254
本来应该很早就写一个的，现在也不晚，慢慢记录吧 1、数据同时存在于CPU和GPU上 报错：RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! 代码片段 class Linear(nn.Module): def __init__(self,in_features=1024,out_features=31, momentum=0.9):
PyTorch基础入门六：PyTorch搭建卷积神经网络实现MNIST手写数字识别
Liam的个人博客
08-05 5万+
1）卷积神经网络(CNN)简介 关于什么是卷积神经网络(CNN)，请自行查阅资料进行学习。如果是初学者，这里推荐一下台湾的李宏毅的深度学习课程。链接就不给了，这些资料网站上随处可见。 值得一提的是，CNN虽然在图像处理的领域具有不可阻挡的势头，但是它绝对不仅仅只能用来图像处理领域，大家熟知的alphaGo下围棋也可以通过CNN的结构进行处理，因为下围棋与图像有着相似之处，所以说，CNN提供给我...
RuntimeError: The size of tensor a (96) must match the size of tensor b (95) at non-singleton dimens
网瘾少年康复中心
03-30 2万+
复现PSMNet时候出现的bug。 出错在output = model(imgL,imgR) 一开始以为左右两图维度不一致。换成output = model(imgR,imgR)之后依旧报错 翻墙查阅pytorch资料 If two tensors x, y are “broadcastable”, the resulting tensor size is calcul
Python_Pytorch_Bug故障处理与学习资料
- 尝试在尽可能简化的环境中复现bug，这有助于定位问题所在，并为bug修复和测试提供便利。 4. 排查Bug - 分析bug可能发生的环节，检查代码逻辑、使用PyTorch提供的调试工具、查看日志文件等方法来逐步排查问题。 ...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值