请教如何在自定义的训练过程中,实现callbacks中的功能呢?
来源:10-10 样例例分析与总结
OliverSong
2019-09-25
请教老师,之前大多的训练不需要自定义步骤,调用.fit()函数就行,fit函数中可以加入callbacks来进行early_stopping之类。
求问如何使自定义训练过程能使用类似callbacks的功能呢?
谢谢!
写回答
1回答
-
正十七
2019-09-26
会有一些API供你调用,比如Tensorboard就有这些API: https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/keras/callbacks/TensorBoard#on_epoch_end
API包括:
on_batch_begin, on_batch_end, on_epoch_begin, on_epoch_end 等等。
需要做的就是初始化这个callback对象,然后在自定义流程的对应位置调用这个对象的相应API。
00
相似问题
为什么每次的训练结果都不一样呢?
回答 2
请教一个交叉验证的问题
回答 1