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

最新推荐文章于 2024-06-04 22:52:33 发布
最新推荐文章于 2024-06-04 22:52:33 发布
阅读量2k 收藏 2
点赞数 2
分类专栏： python 机器学习 文章标签： pytorch 深度学习 神经网络
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_41848012/article/details/120261938
版权

我在训练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(self.parameters()).dtype) # fp16 compatibility

为：

extended_attention_mask = extended_attention_mask.to(dtype=torch.float32) # fp16 compatibility
六六六六神
关注
专栏目录
torch报错：StopIteration: Caught StopIteration in replica 0 on device 0.
weixin_44152453的博客
10-26 7983
这里写自定义目录标题欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题，有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能，丰富你的文章UML 图表FLowchart流程图导出与导入导出导入 欢迎使用Markdown编辑器 你好！ 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Mar
Pytorch Dataloader 模块源码分析（一）：整体框架与 Dataset 组件
weixin_41670608的博客
08-29 807
前文提到 Dataset 类支持两种风格，map 和 iterable，map 风格指的是该 Dataset 对象可以通过传入一个下标（键）来访问具体的值，也就是内部实现是一个哈希表（python 中就是字典），而 Iterable 数据集则是前文提到的可迭代对象，可以通过 for 循环遍历内部的数据。由于篇幅较长，后面的部分将放在后半篇。因此，map 风格的 Dataset 需要实现__getitem__方法而 iterable 风格的 Dataset 需要实现__iter__和__next__方法。..
解决迭代器调用next方法时报错：StopIteration
呆萌的代Ma
07-08 6041
解决方法 当使用next()去访问一个已经迭代完的迭代器时，会有这样的报错：StopIteration 解决方法就是给一个默认值：next(iter , 默认值)，当迭代完成后会输出这个默认值 假设原来循环时的写法是： a = next(iter_test) # 迭代完成会报错StopIteration print(a) 改为： a = next(iter_test,None) if a is not None: print(a) 问题解析 如下代码会报错： iter_list = iter([1,
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:...
SSD-pytorch踩坑及修正（StopIteration）
weixin_36670529的博客
04-23 879
SSD-pytorch踩坑（StopIteration） ssd-pytorch代码链接 bug 解决方案 ssd-pytorch代码链接 github上的：https://github.com/amdegroot/ssd.pytorch bug 这篇博客写的基本上没什么问题，但是其实也有坑。https://blog.csdn.net/sinat_39307513/article/details/84646177 在这篇博客的评论区里面出现了 “博主您好，我训练时候损失降得又快又低，但是运行e
训练Pytorch深度学习模型出现StopIteration
最新发布
weixin_50557558的博客
06-04 327
训练Pytorch深度学习模型出现StopIteration
pytorch dataloader遇到StopIteration问题
moonson96的博客
04-23 2014
pytorch模型训练流程中遇到的一些坑(持续更新)
baidu_41902768的博客
12-29 1万+
要训练一个模型，主要分成几个部分，如下。 数据预处理 入门的话肯定是拿 MNIST 手写数据集先练习。pytorch 中有帮助我们制作数据生成器的模块，其中有 Dataset、TensorDataset、DataLoader 等类可以来创建数据入口。之前在 tensorflow 中可以用 dataset.from_generator() 的形式，pytorch ...
def __next__(self): if self.residue and self.index == self.n_batches: batches = self.batches[self.index * self.batch_size: len(self.batches)] self.index += 1 batches = self._to_tensor(batches) return batches elif self.index >= self.n_batches: self.index = 0 raise StopIteration else: batches = self.batches[self.index * self.batch_size: (self.index + 1) * self.batch_size] self.index += 1 batches = self._to_tensor(batches) return batches
06-02
具体来说，该方法首先判断数据集的 batch 数量是否为整数，如果是并且当前遍历到了最后一个 batch，那么就将 self.index 设置为 0，然后抛出 StopIteration 异常。如果 batch 数量不是整数并且当前遍历到了最后一个 ...
处理StopIteration停止迭代异常
小清华的小哥哥
12-31 5471
Python中迭代器循环遍历完最后一个数据，再用next()方法取值，会抛出异常StopIteration(停止迭代) 解决办法：抓住异常 示例代码： ```python try: # 你的代码段 except StopIteration as s: print('StopIteration')
理解StopIteration的异常
shitoucoming的博客
09-21 7455
在讲循环对象的时候，曾提到一个StopIteration的异常，该异常是在循环对象穷尽所有元素时的报错。 我们以它为例，来说明基本的异常处理。 一个包含异常的程序: a = iter(range(5)) try: for i in range(100): print(a.__next__()) except StopIteration: print('here is end ', i) # 结果展示： ''' 0 1 2 3 4 here is end 5 ''' 首
python生成应用程序错误_“运行时错误：生成器在每次尝试运行应用程序时引发StopIteration”...
weixin_32752813的博客
01-14 527
我试图运行以下代码：import weburls = ('/', 'index')if __name__ == "__main__":app = web.application(urls, globals())app.run()但每次我都会犯这个错误C:\Users\aidke\Desktop>python app.pyTraceback (most recent call last):Fi...
Pytorch报错记录
weixin_30519071的博客
04-20 138
1.BrokenPipeError 执行以下命令时： a,b = iter(train_loader).next() 报错：BrokenPipeError: [Errno 32] Broken pipe 错误原因：子进程原因 On Windows the subprocesses will import (i.e. execute) the main module at s...
StopIteration异常处理分析思想
从0到1
05-16 1万+
案列1 函数中局部代码包含迭代器，在跌倒过程中出现问题 iterable = iter([1,2]) def func(iterable): while True: try: print("ok") print(next(iterable)) print("not ok") e...
关于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) ...
Dataloader使用时iter.next()迭代器出现错误怎么解决以及_MultiProcessingDataLoaderIter报错解决指南
weixin_42785935的博客
01-11 8332
_MultiProcessingDataLoaderIter 以及 多线程出错
Python3.7运行web.py测试时出现RuntimeError: generator raised StopIteration异常
weixin_42077503的博客
04-16 899
解决方法： 参见： https://blog.csdn.net/qq_38591756/article/details/84971485#commentBox
next(self.parameters()).new_zeros()是什么
ccbrid的博客
04-17 5633
Q：next(self.parameters())是什么？ Q：next(self.parameters()).new_zeros()是什么？ Q：next(self.parameters()).data是什么？ Q：next(self.parameters()).data.new(batch_size, self.nhid).zero_()是什么？ A： 1. 使用： 某类的某一函...
pytorch因版本问题出现这个错误怎么解决
05-14
如果你遇到了类似 "AttributeError: module 'torch' has no attribute 'Tensor'" 的错误，可能是因为你使用的PyTorch版本不兼容。解决方法如下： 1.确保你安装的是最新版的PyTorch。可以在PyTorch官网上找到最新版并安装。 2.检查你的代码中是否有导入其他版本的PyTorch的模块。如果有，请将它们删除或注释掉。 3.如果你使用的是Anaconda环境，可以尝试创建一个新的环境，并在其中安装最新版的PyTorch。 4.如果你使用的是pip安装的PyTorch，可以尝试卸载当前版本并重新安装最新版。 5.如果以上方法都无效，可以尝试使用conda安装PyTorch，并使用conda管理你的环境。 希望以上方法能帮到你！
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值