我正在使用预先训练的VGG-16模型进行图像分类。我要添加自定义的最后一层,因为我的分类类别为10。我正在训练200个时代的模型。
我的问题是:如果我在某个时期随机停止(通过关闭python窗口)培训,有什么办法,比如说时期否。 50,然后从那里继续?我已经阅读了有关保存和重新加载模型的信息,但我的理解是,该模型仅适用于我们的自定义模型,而不适用于像VGG-16这样的预训练模型。
答案 0 :(得分:4)
您可以使用ModelCheckpoint
回调定期保存模型。要使用它,请将callbacks
参数传递给fit
方法:
from keras.callbacks import ModelCheckpoint
checkpointer = ModelCheckpoint(filepath='model-{epoch:02d}.hdf5', ...)
model.fit(..., callbacks=[checkpointer])
然后,稍后您可以加载最后保存的模型。有关此回调的更多自定义,请查看文档。
答案 1 :(得分:0)
这是ModelCheckpoint的自定义版本,我用它从给定的时期gist恢复训练。它将纪元和其他日志保存到相应的JSON文件中,还将在开始时检查是否继续训练。您需要调用get_last_epoch
并在initial_epoch
中设置model.fit
才能从那个时期恢复。
import json
class StatefulCheckpoint(ModelCheckpoint):
"""Save extra checkpoint data to resume training."""
def __init__(self, weight_file, state_file=None, **kwargs):
"""Save the state (epoch etc.) along side weights."""
super().__init__(weight_file, **kwargs)
self.state_f = state_file
self.state = dict()
if self.state_f:
# Load the last state if any
try:
with open(self.state_f, 'r') as f:
self.state = json.load(f)
self.best = self.state['best']
except Exception as e: # pylint: disable=broad-except
print("Skipping last state:", e)
def on_train_begin(self, logs=None):
prefix = "Resuming" if self.state else "Starting"
print("{} training...".format(prefix))
def on_epoch_end(self, epoch, logs=None):
"""Saves training state as well as weights."""
super().on_epoch_end(epoch, logs)
if self.state_f:
state = {'epoch': epoch+1, 'best': self.best}
state.update(logs)
state.update(self.params)
with open(self.state_f, 'w') as f:
json.dump(state, f)
def get_last_epoch(self, initial_epoch=0):
"""Return last saved epoch if any, or return default argument."""
return self.state.get('epoch', initial_epoch)