Skip to content
This repository has been archived by the owner on May 8, 2019. It is now read-only.

Latest commit

 

History

History
18 lines (15 loc) · 413 Bytes

SparkNetwork模型存储.md

File metadata and controls

18 lines (15 loc) · 413 Bytes

SparkDl4jMultiLayer模型存储


使用Spark分布式训练需要使用如下的类,那么如何对模型进行保存

SparkDl4jMultiLayer sparkNet = new SparkDl4jMultiLayer(sc, conf, tm);

首先调用getNetwork()方法获取对应的实体类

MultiLayerNetWork network = sparkNet.getNetwork();

然后调用调用

ModelSerializer.restoreMultiLayerNetwork()

对模型进行保存