我想比较不同模型之间的计算时间。
在拟合期间,每个时期的计算时间被打印到控制台。
Epoch 5/5
160000/160000 [==============================] - **10s** ......
我正在寻找一种方法来存储这些时间,其方式与模型指标类似,模型指标保存在每个时期并可通过历史对象获取。
尝试以下回调:
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)
Then:
time_callback = TimeHistory()
model.fit(..., callbacks=[..., time_callback],...)
times = time_callback.times
在这种情况下times
应该存储历元计算时间。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)