关于spark的Deeplearning4j：是否有可能在训练神经网络时捕获每层的权重

Question

我一直试图找到一种方法来捕获和保存（作为json）每层神经网络的权重，同时使用DL4J训练（deeplearning4j on spark）。

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

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。