请教如何在自定义的训练过程中,实现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。

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程