我正在一个使用Keras ModelCheckpoint
的项目中工作。这个回调类似乎可以满足我的需求,除了一个小细节。
我还没有找到一种将计数器传递给纪元编号以处理履历模型训练案例的方法。我经常训练一些纪元,然后再恢复训练。我想要一个一致的模型保存模式,例如:
model.{epoch:03d}-{loss:.2f}.hdf5
但从纪元开始编号,先前的训练停止了,而不是从头开始。
我当前使用的命令是这样:
ckp_saver = ModelCheckpoint(checkpoint_dir + "/model.{epoch:03d}-{loss:.2f}.hdf5", monitor='loss', verbose=0,
save_best_only=False, save_weights_only=True, mode='auto', period=1)
有什么办法可以将此信息传递给ModelCheckpoint
?我找到的解决方案是编辑Keras代码并添加包含实际预训练历元的默认参数(如果未通过,则默认为0
),以免破坏其他任何代码,但如果不这样做,我希望避免这样做必要。还有其他想法吗?
原始代码来自此文件here。