org.deeplearning4j.core.listener

Class SystemInfoFilePrintListener

java.lang.Object

org.deeplearning4j.core.listener.SystemInfoFilePrintListener

All Implemented Interfaces:

TrainingListener

public class SystemInfoFilePrintListener
extends Object
implements TrainingListener

Using SystemInfo - it prints a json representation on each callback to the specified file.

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
SystemInfoFilePrintListener(boolean printOnEpochStart, boolean printOnEpochEnd, boolean printOnForwardPass, boolean printOnBackwardPass, boolean printOnGradientCalculation, @NonNull File printFileTarget)

Method Summary

Modifier and Type	Method and Description
void	iterationDone(Model model, int iteration, int epoch)
void	onBackwardPass(Model model)
void	onEpochEnd(Model model)
void	onEpochStart(Model model)
void	onForwardPass(Model model, List<INDArray> activations)
void	onForwardPass(Model model, Map<String,INDArray> activations)
void	onGradientCalculation(Model model)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SystemInfoFilePrintListener

public SystemInfoFilePrintListener(boolean printOnEpochStart,
                                   boolean printOnEpochEnd,
                                   boolean printOnForwardPass,
                                   boolean printOnBackwardPass,
                                   boolean printOnGradientCalculation,
                                   @NonNull
                                   @NonNull File printFileTarget)

Method Detail

iterationDone

public void iterationDone(Model model,
                          int iteration,
                          int epoch)

Specified by:

iterationDone in interface TrainingListener

onEpochStart

public void onEpochStart(Model model)

Specified by:

onEpochStart in interface TrainingListener

onEpochEnd

public void onEpochEnd(Model model)

Specified by:

onEpochEnd in interface TrainingListener

onForwardPass

public void onForwardPass(Model model,
                          List<INDArray> activations)

Specified by:

onForwardPass in interface TrainingListener

onForwardPass

public void onForwardPass(Model model,
                          Map<String,INDArray> activations)

Specified by:

onForwardPass in interface TrainingListener

onGradientCalculation

public void onGradientCalculation(Model model)

Specified by:

onGradientCalculation in interface TrainingListener

onBackwardPass

public void onBackwardPass(Model model)

Specified by:

onBackwardPass in interface TrainingListener