ai.djl.training.listener

Class TimeMeasureTrainingListener

java.lang.Object

ai.djl.training.listener.TrainingListenerAdapter

ai.djl.training.listener.TimeMeasureTrainingListener

All Implemented Interfaces:

TrainingListener

public class TimeMeasureTrainingListener
extends TrainingListenerAdapter

TrainingListener that outputs the training time metrics after training is done.

The training time data is placed in the file "$outputDir/training.log" and the validation data is placed in "$outputDir/validate.log".

Nested Class Summary

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

TrainingListener.BatchData, TrainingListener.Defaults

Constructor Summary

Constructors

Constructor and Description
TimeMeasureTrainingListener(java.lang.String outputDir) Constructs a TimeMeasureTrainingListener.

Method Summary

Modifier and Type	Method and Description
void	onEpoch(Trainer trainer) Listens to the end of an epoch during training.
void	onTrainingBatch(Trainer trainer, TrainingListener.BatchData batchData) Listens to the end of training one batch of data during training.
void	onTrainingEnd(Trainer trainer) Listens to the end of training.
void	onValidationBatch(Trainer trainer, TrainingListener.BatchData batchData) Listens to the end of validating one batch of data during validation.

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

onTrainingBegin

Methods inherited from class java.lang.Object

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

Constructor Detail

TimeMeasureTrainingListener

public TimeMeasureTrainingListener(java.lang.String outputDir)

Constructs a TimeMeasureTrainingListener.

Parameters:

outputDir - the directory to output the tracked timing data in

Method Detail

onEpoch

public void onEpoch(Trainer trainer)

Listens to the end of an epoch during training.

Specified by:

onEpoch in interface TrainingListener

Overrides:

onEpoch in class TrainingListenerAdapter

Parameters:

trainer - the trainer the listener is attached to

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

onValidationBatch

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

Listens to the end of validating one batch of data during validation.

Specified by:

onValidationBatch in interface TrainingListener

Overrides:

onValidationBatch in class TrainingListenerAdapter

Parameters:

trainer - the trainer the listener is attached to

batchData - the data from the batch

onTrainingEnd

public void onTrainingEnd(Trainer trainer)

Listens to the end of training.

Specified by:

onTrainingEnd in interface TrainingListener

Overrides:

onTrainingEnd in class TrainingListenerAdapter

Parameters:

trainer - the trainer the listener is attached to