public class SystemInfoPrintListener extends Object implements TrainingListener
SystemInfo - it logs a json representation
of system info using slf4j.| Constructor and Description |
|---|
SystemInfoPrintListener() |
| 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) |
public void iterationDone(Model model, int iteration, int epoch)
iterationDone in interface TrainingListenerpublic void onEpochStart(Model model)
onEpochStart in interface TrainingListenerpublic void onEpochEnd(Model model)
onEpochEnd in interface TrainingListenerpublic void onForwardPass(Model model, List<INDArray> activations)
onForwardPass in interface TrainingListenerpublic void onForwardPass(Model model, Map<String,INDArray> activations)
onForwardPass in interface TrainingListenerpublic void onGradientCalculation(Model model)
onGradientCalculation in interface TrainingListenerpublic void onBackwardPass(Model model)
onBackwardPass in interface TrainingListenerCopyright © 2020. All rights reserved.