Package ai.djl.training.listener

Class DivergenceCheckTrainingListener

java.lang.Object

ai.djl.training.listener.TrainingListenerAdapter

ai.djl.training.listener.DivergenceCheckTrainingListener

All Implemented Interfaces:

TrainingListener

public class DivergenceCheckTrainingListener
extends TrainingListenerAdapter

TrainingListener that gives early warning if your training has failed by divergence.

Nested Class Summary

Nested classes/interfaces inherited from interface ai.djl.training.listener.TrainingListener

TrainingListener.BatchData, TrainingListener.Defaults

Constructor Summary

Constructors

Constructor	Description
DivergenceCheckTrainingListener()

Method Summary

Modifier and Type	Method	Description
void	onTrainingBatch(Trainer trainer, TrainingListener.BatchData batchData)	Listens to the end of training one batch of data during training.

Methods inherited from class ai.djl.training.listener.TrainingListenerAdapter

onEpoch, onTrainingBegin, onTrainingEnd, onValidationBatch

Methods inherited from class java.lang.Object

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

Constructor Detail

DivergenceCheckTrainingListener

public DivergenceCheckTrainingListener()

Method Detail

onTrainingBatch

public void onTrainingBatch(Trainer trainer,
                            TrainingListener.BatchData batchData)

Listens to the end of training one batch of data during training.

Specified by:

onTrainingBatch in interface TrainingListener

Overrides:

onTrainingBatch in class TrainingListenerAdapter

Parameters:

trainer - the trainer the listener is attached to

batchData - the data from the batch