Keras:ModelCheckpoint是保存到变量而不是文件吗?

时间:2019-01-16 08:28:30

标签: python keras neural-network

我想针对X个时期训练我的模型,并以最小的验证误差获得最佳结果。

我发现的解决方案是使用ModelCheckpoint回调,如果当前时期的结果更好,可以将其配置为覆盖模型保存。

但是,在最坏的情况下,这意味着在每个时期都会将模型写入硬盘驱动器,这会产生巨大的时间开销。

是否可以将模型保存到变量中,所以我只需要保存一次?还是我可以做另一种优化?

1 个答案:

答案 0 :(得分:0)

根据save_model的文档字符串,该路径也可以是h5py.File对象,并且h5py文档中有an example of in-memory files