当前位置: 代码网 > it编程>游戏开发>ar > WARNING:tensorflow:Your input ran out of data; interrupting training. 解决方法

WARNING:tensorflow:Your input ran out of data; interrupting training. 解决方法

2024年08月05日 ar 我要评论
WARNING:tensorflow:Your input ran out of data; interrupting training. Make sure that your dataset or generator can generate at least steps_per_epoch * epochs batches (in this case, 13800 batches). You may need to use the repeat() function when building you

问题详情:

根据报错信息,问题似乎出现在数据集生成器上。可以使用数据集的repeat()函数来确保生成足够的批次供训练使用。

我们可以对训练数据集(train_dataset)应用repeat()函数,以确保生成足够的批次:

train_dataset = train_dataset.repeat()

问题原因:

generator产生的训练数据不够用,少于所要求的steps_per_epoch * epochs 个batch数。
(len(train_data)/batch_size)*epochs=所需要的训练数据的batch数
如果盲目的按照提示repeat()函数来扩充train_data的话,要求的所需的batch数也会相应地增多,会陷入一条死路。。

具体修改如下:

train_dataset = build_dataset(x_train, y_train, bsize=batch_size, decode_fn=train_decoder, 
                            augmentadv=false, augment=false, augmentadvseg=true)
train_dataset = train_dataset.repeat()

这样,在开始训练之前,数据集将使用 repeat() 函数来确保生成足够的批次供训练使用。 

之前两个epoch就会停止,现在可以正常运行了

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com