torchkeras怎么按step保存

[terminator123]

还有个问题想请教下，你知道torchkeras怎么按n个step进行报错吗，大模型跑一遍epoch太久了，有时候中间step的结果更好

[necrophagists]

kerasmodel.fit()加个参数patience=X 就是加入早停机制,默认是验证loss如果连续X个step没有下降就停止；如果你要实现n个step保存的话建议看一下torchkeras里调用save_ckpt的部分并修改，应该是加个步数的判断就行了。

[terminator123]

kerasmodel.fit()加个参数patience=X 就是加入早停机制,默认是验证loss如果连续X个step没有下降就停止；如果你要实现n个step保存的话建议看一下torchkeras里调用save_ckpt的部分并修改，应该是加个步数的判断就行了。

patience=x，这个是x个epoch吧。。。save_ckpt那部分我看了下，一是没有step这个参数，另外是我没找到它实际的调用入口啊

[necrophagists]

抱歉 确实是epoch 然后我看了一下 https://github.com/lyhue1991/torchkeras/blob/master/torchkeras/kerasmodel.py 我觉得你可以把KerasModel里的save_ckpt方法传入EpochRunner 然后在EpochRunner中加个判断就行了 EpochRunner有step