使用next(iter(data.DataLoader())报错StopIteration

使用next(iter(data.DataLoader())报错StopIteration，这是因为当使用next()去访问一个已经迭代完的迭代器时，就会有触发这样的报错：StopIteration，即dataloader导入数据之后经过了一轮的迭代，再次导入的时候发现没有数据了，也就是iterable完成之后，触发了StopIteration，然后跳出了循环

解决方法：

既然再次导入的时候没有数据了，那我们就再用一个dataloader就行了。

将train.py中的

inps, targets = next(self.batch_iterator)

更改成：

try:inps, targets = next(self.batch_iterator)
except StopIteration:self.batch_iterator = iter(data.DataLoader(self.train_dataset, self.args.batch_size, shuffle=True, num_workers=self.args.num_workers, collate_fn=detection_collate))inps, targets = next(self.batch_iterator)

问题解决。

使用next(iter(data.DataLoader())报错StopIteration相关推荐

mysql load报错_mysql:执行LOAD DATA LOCAL 报错
mysql:执行LOAD DATA LOCAL 报错. 我使用navicat for mysql 连接 mysql服务器执行local 可以正常执行 . show VARIABLES like '% ...
使用客户端jedis时报错Could not get a resource from the pool 以及使用Spring Data Redis报错解决方法
一.Jedis 报错今天在使用jedis时,一直报错 Could not get a resource from the pool 在网上找了好多解决的方法,并且找了半天错误,才发现是我的启动方式有 ...
【Unity导入MySql.Data.dll报错】
Unity导入MySql.Data.dll报错错误的起因,想用Unity对MySQL操作.操作参考了勤诚勇毅的帖子链接: https://blog.csdn.net/qq_43333567/art ...
【数据库】【Mysql - Workbench1】：如何解决WPS/excel数据导入时Table Data Import报错问题？
总结 1.Mysql Workbench导入WPS.Excel的csv.文件出错[Table Data Import]是因为utf-8编码不匹配,因此,只需要把目标csv.文件编码设置成utf-8即可 ...
mysql建表指定引擎_请教mysql建表指定data directory 报错
-- 如下: 语句 1... create table test(id int) engine=myisam data directory='/tmp' index directory='/tmp/a ...
Pytorch_Geometric（PyG）使用DataLoader报错RuntimeError: Sizes of tensors must match except in dimension 0.
使用Pytorch_Geometric(PyG)时构建DataLoader,从DataLoader获取样本Batch时报错:RuntimeError: Sizes of tensors must ma ...
FlowableException: Error initialising dmn data model报错问题
flowable war包启动失败,报错Caused by: org.flowable.common.engine.api.FlowableException: Error initialising ...
IDEA lombok包的@Data注解报错找不到符号
问题描述: 工程编译报错 java:找不到符号问题抽象定位错误可以发现是getter setter的问题由于采用了@Data注解,所以优先考虑是lombok的问题解决方案首先排除以下可能 l ...
解决 uboot data abort报错
新手解决bug 一步一坑-_-! 一切从最基础的开始加油报错内容如下: 536 bytes read in 6 ms (86.9 KiB/s) ** File not found /boot/dt ...

使用next(iter(data.DataLoader())报错StopIteration

使用next(iter(data.DataLoader())报错StopIteration相关推荐

最新文章

热门文章