Keras 警告 on_batch_begin

Question

在使用回调 ResetStatesCallback 训练 rnn 模型时，我收到以下警告消息：

/var/venv/DSTL/lib/python3.4/site-packages/keras/callbacks.py:97： UserWarning：与批处理相比，方法 on_batch_begin() 速度较慢 更新（0.791834）。检查你的回调。 % delta_t_median)

from keras.callbacks import Callback
#Reset count every RESET_STATES_LENGTH
#RESET_STATES_LENGTH=8
class ResetStatesCallback(Callback):
    def __init__(self):
        self.counter = 0

    def on_batch_begin(self, batch, logs={}):
        if self.counter % RESET_STATES_LENGTH == 0:
            self.model.reset_states()
        self.counter += 1

为什么我会收到此消息？我应该尝试解决它吗？它真的会减慢我的训练速度吗？

Answer 1

请参阅https://github.com/fchollet/keras/issues/5008 了解说明。上面写着

您正在运行诸如在每批之后保存模型或渲染图像之类的操作，并且它所花费的时间比批次本身要长。

因此，Keras 在运行时似乎已确定您的回调比批处理本身慢。