第一篇
承接上文
在搭载好环境之后，满心欢喜准备搭载模型的绘图（损失曲线、验证精度）功能，我就先使用10个epoch跑一下，结果。
它卡在199/200了！之前从没遇到过这个情况。
接着跑，还是不成功。
网上很多报错都是pytorch版本得dataloader，不太能够适配。我就开始自身琢磨，我发现训练过程是没问题得，那么问题是出在我的valmodel，最后一步savemodel根本没走过去（QAQ深度学习不会打断点啊！）。
那么我的数据集已经成功导入了，为什么我没有继续前进呢？原因如下：
[En]
So my dataset was successfully imported, why didn’t I move on? The reasons are as follows:
我一开始设置的history_fit中的steps_per_epoch大于validation_steps，导致训练完模型之后喂给模型的数据太多了导致OOM，内存溢出。
接下来就是把validation_steps调小即可。
如果你的steps_per_epoch为200，那么validation_steps就应该是100或者更小（不建议）。最后就成功啦！
关于history_fit中的steps_per_epoch大于validation_steps，想深入学习的可以点开下面两个链接。
steps_per_epoch和epoch如何设置
源码分析 steps_per_epoch 参数
以上。
Original: https://blog.csdn.net/Dcurry/article/details/121466819
Author: Dcurry
Title: 深度学习Tensorflow的坑（二：训练一个epoch卡住不动，程序不终止也不继续）
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/496841/
转载文章受原作者版权保护。转载请注明原作者出处！