java - 深度学习4j : update a saved model

标签 java deep-learning deeplearning4j

我使用 deeplearning4j 在数据的第一部分上训练了一个模型并保存了它。

ModelSerializer.writeModel(model, locationToSave, true);

现在,我想在下一部分数据上训练这个保存的模型。

我加载了模型:

MultiLayerNetwork model = ModelSerializer.restoreMultiLayerNetwork("location");

然后我用新数据来训练它

model.fit(trainingDataIt);

但是模型似乎不合适,因为我没有看到每次迭代的统计数据

11:56:56.161 [ADSI prefetch thread] DEBUG o.n.l.memory.abstracts.Nd4jWorkspace - Steps: 4
11:59:30.072 [main] DEBUG o.d.d.iterator.AsyncDataSetIterator - Manually destroying ADSI workspace
11:59:30.123 [ADSI prefetch thread] DEBUG o.n.l.memory.abstracts.Nd4jWorkspace - Steps: 4
12:01:39.760 [main] DEBUG o.d.d.iterator.AsyncDataSetIterator - Manually destroying ADSI workspace
12:01:39.793 [ADSI prefetch thread] DEBUG o.n.l.memory.abstracts.Nd4jWorkspace - Steps: 4
12:03:46.496 [main] DEBUG o.d.d.iterator.AsyncDataSetIterator - Manually destroying ADSI workspace
12:03:46.551 [ADSI prefetch thread] DEBUG o.n.l.memory.abstracts.Nd4jWorkspace - Steps: 4

这是正常现象,还是我的方法有问题?

最佳答案

你只需调用 ModelSerializer.writeModel 加载它并调用 fit 之后。

关于java - 深度学习4j : update a saved model,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52403395/

相关文章:

java - 从不同的 gradle 模块导入类

java - 创建扩展抽象类的子类的实例 (Java)

scikit-learn - sklearn、Keras、DeepStack - ValueError : multi_class must be in ('ovo' , 'ovr' )

Tensorflow 2.0将nlp的预处理tonkezier保存到tensorflow服务器中

tensorflow - CNN 在 keras 中具有多个 conv3d

java - 运行 deeplearning4J 示例时出现问题

Java,在整数列表中找到交集

java - CSVRecordReader 和 CSV 行末尾未终止的引用字段

java - deeplearning4j - 使用 Word2Vec 进行命名实体识别

java - 如何将命令添加到 Ant 构建 xml