Keras警告on_batch_begin

Question

当使用回调ResetStatesCallback bellow训练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已经确定您的回调比批处理本身慢。