在model.fit（）

Question

記錄Keras中每個曆元的計算時間我想比較不同模型之間的計算時間。 在擬合期間，每個時期的計算時間被打印到控制檯。

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

我正在尋找一種方式將這些時間存儲在一個相似的方式保存在每個時間段，並通過歷史記錄對象avaliable模型指標。

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應存儲時代的計算時間。