如何在指定的纪元数结束时获取回调？

Question

我想在使用Keras时对模型进行微调，我想要更改训练数据和学习率，以便在纪元数达到10时进行训练，那么如何在指定的纪元数结束时获得回调。

Answer 1

您需要编写自己的回调子类。

https://keras.io/callbacks/ (一般信息)

https://github.com/keras-team/keras/blob/master/keras/callbacks.py#L275 (回调基类的源代码)

您的回调子类应该定义一个on_epoch_end()方法，该方法接受纪元号作为参数。

Answer 2

实际上，keras的工作方式可能不是最好的方式，最好将其视为微调，这意味着您完成了10个时期，保存模型，然后加载模型(从另一个脚本)，并继续使用lr和您喜欢的数据进行训练。

这有几个原因。

1. 它更清晰，更容易调试。在10个时期之后，你检查你的模型是否正确，验证它是否工作正常，并继续进行
2. 。从10时期开始，用这种方法做几个实验要好得多。

祝好运!