当使用回调ResetStatesCallback bellow训练rnn模型时,我收到以下警告消息:
/var/venv/DSTL/lib/python3.4/site-packages/keras/callbacks。py:97:
UserWarning:u batch\u begin()上的方法与批处理相比速度较慢
更新(0.791834)。检查您的回电。%delta\u t\u中值)
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
为什么我会收到这个消息?我应该试着修一下吗?这真的会让我的训练慢那么多吗?
