python - Keras：ModelCheckpoint 保存到变量而不是文件？

Question

我想为 X 时期训练我的模型，并在最小验证错误方面获得最佳结果。

我找到的解决方案是使用 ModelCheckpoint 回调，如果当前 epoch 的结果更好，可以将其配置为覆盖模型保存。

然而，在最坏的情况下，这意味着在每个 epoch 中都会将一个模型写入硬盘驱动器，这会产生巨大的时间开销。

是否有可能将模型保存到变量中，所以我只需要保存一次？或者我可以做其他优化吗？

标签： pythonkerasneural-network

Answer 1

根据save_model的文档字符串，路径也可以是一个h5py.File对象，在 h5py 文档中有一个内存文件的例子。