我正在使用较小的数据集和模型,而我正在尝试开发的算法依赖于模型权重的快速保存和加载。因为我的尺寸很小,所以不受内存限制。
对我来说,主要的速度瓶颈是在训练期间将权重保存到#include <ncurses.h>
文件中所需的时间。由于我不知道的某些原因,这仅在第一个时期发生。
为什么在设置模型检查点时的第一个时期存在这种滞后?我有什么可以做的吗?有什么方法可以将权重作为数组列表转储到内存中?从我的角度来看,那是最简单的。
这是MWE:
hdf5
在任何人说什么之前:我知道,在python上对Keras有更多支持。如果您知道的话,将喜欢用该语言编写的示例。