标签: python keras neural-network
我想针对X个时期训练我的模型,并以最小的验证误差获得最佳结果。
我发现的解决方案是使用ModelCheckpoint回调,如果当前时期的结果更好,可以将其配置为覆盖模型保存。
但是,在最坏的情况下,这意味着在每个时期都会将模型写入硬盘驱动器,这会产生巨大的时间开销。
是否可以将模型保存到变量中,所以我只需要保存一次?还是我可以做另一种优化?
答案 0 :(得分:0)
根据save_model的文档字符串,该路径也可以是h5py.File对象,并且h5py文档中有an example of in-memory files。
h5py.File