如何在模型检查点期间加快权重的节省?在第一个时期,callback_model_checkpoint确实很慢

时间:2018-06-19 23:25:40

标签: r tensorflow io keras

我正在使用较小的数据集和模型,而我正在尝试开发的算法依赖于模型权重的快速保存和加载。因为我的尺寸很小,所以不受内存限制。

对我来说,主要的速度瓶颈是在训练期间将权重保存到#include <ncurses.h>文件中所需的时间。由于我不知道的某些原因,这仅在第一个时期发生。

为什么在设置模型检查点时的第一个时期存​​在这种滞后?我有什么可以做的吗?有什么方法可以将权重作为数组列表转储到内存中?从我的角度来看,那是最简单的。

这是MWE:

hdf5

在任何人说什么之前:我知道,在python上对Keras有更多支持。如果您知道的话,将喜欢用该语言编写的示例。

0 个答案:

没有答案