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 для получения дополнительной информации.

Другие вопросы по тегам