は、model.fit（）中のKerasの各エポックの計算時間を記録します。

Question

異なるモデル間の計算時間を比較したいと思います。 フィット時に、1エポックあたりの計算時間がコンソールに出力されます。

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

私は歴史のオブジェクトを介して各エポックとavaliableに保存されているモデルの指標と同様にこれらの時間を格納するための方法を探しています。エポック計算時間を格納する必要があります。この場合timesで

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

：次に

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) 

：

Answer 1

は、次のコールバックを試してみてください。