apache-spark - Spark 上的 Deeplearning4j：是否可以在训练神经网络时捕获每一层的权重

Question

我一直在尝试寻找在使用 DL4J（deeplearning4j on spark）进行训练时是否有办法捕获和保存（作为 json 格式）神经网络每一层的权重。

如果有人有任何想法，请告诉我。

Answer 1

在训练循环期间，您从 fit 中获取多层网络： https ://github.com/deeplearning4j/deeplearning4j/blob/master/deeplearning4j-scaleout/spark/dl4j-spark/src/main/java/org/deeplearning4j/spark/ impl/multilayer/SparkDl4jMultiLayer.java#L202

这是模型的本地版本，你可以随心所欲。有关更多信息，请参见http://deeplearning4j.org/spark。