在model.fit（）期间，在Keras中记录每个历元的计算时间

Question

我想比较不同模型之间的计算时间。 在拟合期间，每个历元的计算时间将打印到控制台。

Epoch 5/5
160000/160000 [==============================] - **10s** ......

我正在寻找一种存储这些时间的方法，这种方法与保存在每个纪元中并且可以通过history对象获得的模型度量相似。

Answer 1

尝试以下回调：

class TimeHistory(keras.callbacks.Callback):
    def on_train_begin(self, logs={}):
        self.times = []

    def on_epoch_begin(self, batch, logs={}):
        self.epoch_time_start = time.time()

    def on_epoch_end(self, batch, logs={}):
        self.times.append(time.time() - self.epoch_time_start)

然后：

time_callback = TimeHistory()
model.fit(..., callbacks=[..., time_callback],...)
times = time_callback.times

在这种情况下times应该存储epoch计算时间。