Deeplearning4j on spark: возможно ли фиксировать веса для каждого слоя во время тренировки нейронной сети
Я пытался найти, есть ли способ для захвата и сохранения (как json) весов каждого слоя нейронной сети во время тренировки с использованием DL4J(deeplearning4j на искре).
Пожалуйста, дайте мне знать, если у кого-то есть идеи.
1 ответ
Во время учебного цикла вы возвращаете многослойную сеть из нужного: 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 для получения дополнительной информации.