public class DivergenceCheckTrainingListener extends TrainingListenerAdapter
TrainingListener
that gives early warning if your training has failed by divergence.TrainingListener.BatchData, TrainingListener.Defaults
Constructor and Description |
---|
DivergenceCheckTrainingListener() |
Modifier and Type | Method and Description |
---|---|
void |
onTrainingBatch(Trainer trainer,
TrainingListener.BatchData batchData)
Listens to the end of training one batch of data during training.
|
onEpoch, onTrainingBegin, onTrainingEnd, onValidationBatch
public void onTrainingBatch(Trainer trainer, TrainingListener.BatchData batchData)
onTrainingBatch
in interface TrainingListener
onTrainingBatch
in class TrainingListenerAdapter
trainer
- the trainer the listener is attached tobatchData
- the data from the batch