训练Pytorch深度学习模型出现StopIteration

Andrew_Xzw

于 2024-06-04 22:52:33 发布

阅读量328

点赞数 3

文章标签：深度学习 pytorch 人工智能

本文链接：https://blog.csdn.net/weixin_50557558/article/details/139455964

版权

训练一个深度学习检测模型，突然出现：
在这里插入图片描述
是因为next(batch_iterator)，可能迭代器读出来的数据为空。

 # load train data
 # 原先代码
 images, targets = next(batch_iterator)
 # 更改为：
 try:
     images, targets = next(batch_iterator)
 except:
     batch_iterator = iter(data_loader)
     images, targets = next(batch_iterator)

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

Andrew_Xzw

关注关注

3
点赞
踩
0

收藏

觉得还不错? 一键收藏
打赏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

今日报错记录：StopIteration、指定显卡、lines = file.readlines()

qq_60074111的博客

01-25

1150

使用单gpu的时候是正常的，但是使用多gpu的时候会报错。问题是多gpu进行模型训练的时候产生的，具体为，不能够用多gpu加载预训练的bert。应该是torch版本的问题。根据可以知道，torch1.5版本有这个问题，我是torch1.6也有这个问题，据替换为torch1.4可以解决该问题。

Pytorch用next()的时候抛出StopIteration的错误的解决方法

六六六六神的博客

09-13

2017

我在训练vilbert的时候，报错如下： File "/vilbert/vilbert.py", line 1351, in forward dtype=next(self.parameters()).dtype StopIteration 查了一下可能是pytorch版本问题，所以有两种解决方法：解决方法一降级pytorch==1.4.0 解决方法二修改代码： extended_attention_mask = extended_attention_mask.to(dtype=next

参与评论您还未登录，请先登录后发表或查看评论

Pytorch运行CNN代码读取数据出现StopIteration的异常及解决方案

qq_28424679的博客

04-07

2058

如果只是单纯的依靠 next(trainloader_iter) 获取数据，则有可能获取数据时候出现不能获取的情况数据为non，此时我们需要再次回访该数据集，再次肚读取。代码如下： #读取数据代码部分： train_dataset = Dataset(args.data_dir, args.data_list, mean=IMG_MEAN) train_dataset_size = len(train_dataset) print ('dataset size:...

关于StopIteration的处理

坑底Z蛙

06-15

1万+

def coroutine(func): def inner(*args,**kwargs): f=func(*args,**kwargs) next(f) return f return inner@coroutine def grep(pattern): print("Looking for %s"%pattern) ...

pytorch dataloader遇到StopIteration问题

moonson96的博客

04-23

2015

一文弄懂Pytorch的DataLoader,Dataset,Sampler之间的关系

最新发布

kuxingseng123的博客

12-31

1048

慢慢的将各种数据之间的关系全部都搞透彻,将其研究彻底.

[PyTorch 学习笔记] 2.1 DataLoader 与 DataSet

张贤的博客

08-25

973

本章代码：https://github.com/zhangxiann/PyTorch_Practice/blob/master/lesson2/rmb_classification/ 人民币二分类实现 1 元人民币和 100 元人民币的图片二分类。前面讲过 PyTorch 的五大模块：数据、模型、损失函数、优化器和迭代训练。数据模块又可以细分为 4 个部分：数据收集：样本和标签。数据划分：训练集、验证集和测试集数据读取：对应于PyTorch 的 DataLoader。其中 DataLoad.

成功运行SSD_Pytorch并训练自己的数据集

m0_47452894的博客

01-18

1万+

目录前言一、环境二、准备数据集1.voc2007数据集2.voc2012数据集三、从Github上下载作者源码创建文件夹四、修改代码1.修改data/config.py2.修改data/voc0712.py3.修改layers/modules/multibox_loss.py4.修改train.py5.修改ssd.py五、开始训练六、运行eval.py1.修改eval.py2.修改ssd.py七、运行test.py修改test.py八、运行demo.ipynb修改demo.py总结前言本文主要是记录自己

SSD-Pytorch：更紧凑，更简单的SSD pytorch实现

02-14

通常，一个深度学习项目的结构会包括数据预处理、模型定义、训练循环、验证和测试等部分。在这个SSD-Pytorch实现中，每个部分都应该被设计得清晰明了，使得新手也能快速上手。数据集的定制是另一个关键特性。在...

torch报错：StopIteration: Caught StopIteration in replica 0 on device 0.

weixin_44152453的博客

10-26

7988

这里写自定义目录标题欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题，有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能，丰富你的文章UML 图表FLowchart流程图导出与导入导出导入欢迎使用Markdown编辑器你好！这是你第一次使用 Markdown编辑器所展示的欢迎页。如果你想学习如何使用Mar

window配置cpu深度学习环境 keras+tensorflow+torch 版本搭配详细教程 pip 下载sklearn

11-14

3859

python配置cpu深度学习环境 keras+tensorflow+torch+bert_wwm_ext_pytorch 版本搭配

SSD-pytorch踩坑及修正（StopIteration）

打工人陈墨的博客

09-13

8305

SSD-pytorch踩坑ssd-pytorch代码链接bug解决方案 ssd-pytorch代码链接 github上的：https://github.com/amdegroot/ssd.pytorch bug 这篇博客写的基本上没什么问题，但是其实也有坑。 https://blog.csdn.net/sinat_39307513/article/details/84646177 在这篇博客的评论...

PyTorch源码解读之torch.utils.data.DataLoader

热门推荐

AI之路

01-14

22万+

PyTorch中数据读取的一个重要接口是torch.utils.data.DataLoader，该接口定义在dataloader.py脚本中，只要是用PyTorch来训练模型基本都会用到该接口，该接口主要用来将自定义的数据读取接口的输出或者PyTorch已有的数据读取接口的输入按照batch size封装成Tensor，后续只需要再包装成Variable即可作为模型的输入，因此该接口有点承上启下的

【pytorch源码赏析】Dataset in pytorch

Duino的工具箱

08-13

6677

Dataset in pytorch

Pytorch数据读取(Dataset, DataLoader, DataLoaderIter)

weixin_39739616的博客

11-07

5677

Pytorch的数据读取主要包含三个类: Dataset DataLoader DataLoaderIter 这三者大致是一个依次封装的关系: 1.被装进2., 2.被装进3. 一. torch.utils.data.Dataset 是一个抽象类, 自定义的Dataset需要继承它并且实现两个成员方法: __getitem__() __len__() 第一个最为重要, 即每次怎...

学习笔记|Pytorch使用教程30(PyTorch常见报错 )

qq_24739717的博客

11-28

2301

学习笔记|Pytorch使用教程30 本学习笔记主要摘自“深度之眼”，做一个总结，方便查阅。使用Pytorch版本为1.2

PyTorch学习总结(四)——Utilities

码农王小呆的博客

12-05

3319

1. PackedSequence torch.nn.utils.rnn.PackedSequence 这个类的实例不能手动创建。它们只能被pack_padded_sequence() 实例化。2. pack_padded_sequence torch.nn.utils.rnn.pack_padded_sequence()** 输入：input: [seq_length x batch_si

深度学习模型训练：PyTorch优化器与超参数配置

"该资源是关于深度学习模型训练的PPT，主要以PyTorch为例，涵盖了模型训练的超参数设定、多GPU训练策略、训练可视化，并提供了代码实例。" 在深度学习领域，模型训练是核心步骤，它涉及到多个关键组成部分。首先，...