神经网络的保存和加载

ModelSerializer是用于加载和保存模型的类。保存的方法有两种(见以下示例): 第一个示例保存的是普通的多层网络,第二个示例保存的是一个计算图

下面这个基本示例给出了用ModelSerializer类保存计算图所需的代码,另一个示例说明了如何用ModelSerializer保存用MultiLayer Configuration构建的神经网络。


RNG种子

如果您的模型用到概率(即DropOut/DropConnect),那么最好能将其分开保存,待模型还原后再加以应用。即:


Nd4j.getRandom().setSeed(12345);
 ModelSerializer.restoreMultiLayerNetwork(modelFile);

这将确保不同会话/JVM能产生一致的结果。

与我们在Gitter聊天