ai.djl.training.listener

Interface TrainingListener.Defaults

Enclosing interface:

TrainingListener

public static interface TrainingListener.Defaults

Contains default TrainingListener sets.

Method Summary

Modifier and Type	Method and Description
static TrainingListener[]	logging(java.lang.String name, int batchSize, int trainDataSize, int validateDataSize, java.lang.String outputDir) A default TrainingListener set including batch output logging.

Method Detail

logging

static TrainingListener[] logging(java.lang.String name,
                                  int batchSize,
                                  int trainDataSize,
                                  int validateDataSize,
                                  java.lang.String outputDir)

A default TrainingListener set including batch output logging.

Parameters:

name - the name of the LoggingTrainingListener

batchSize - the size of training batches

trainDataSize - the total number of elements in the training dataset

validateDataSize - the total number of elements in the validation dataset

outputDir - the output directory to store created log files

Returns:

the new set of listeners