Package ai.djl.training.listener
Interface TrainingListener.Defaults
-
- Enclosing interface:
- TrainingListener
public static interface TrainingListener.Defaults
Contains defaultTrainingListener
sets.
-
-
Method Summary
Static Methods Modifier and Type Method Description static TrainingListener[]
basic()
A basicTrainingListener
set with minimal recommended functionality.static TrainingListener[]
logging()
A defaultTrainingListener
set including batch output logging.static TrainingListener[]
logging(int frequency)
A defaultTrainingListener
set including batch output logging.static TrainingListener[]
logging(java.lang.String outputDir)
A defaultTrainingListener
set including batch output logging and output directory.
-
-
-
Method Detail
-
basic
static TrainingListener[] basic()
A basicTrainingListener
set with minimal recommended functionality.This contains:
- Returns:
- the new set of listeners
-
logging
static TrainingListener[] logging()
A defaultTrainingListener
set including batch output logging.This contains:
- Everything from
basic()
LoggingTrainingListener
- Returns:
- the new set of listeners
- Everything from
-
logging
static TrainingListener[] logging(int frequency)
A defaultTrainingListener
set including batch output logging.This has the same listeners as
logging()
, but reduces the logging frequency.- Parameters:
frequency
- the frequency of epoch to print out- Returns:
- the new set of listeners
-
logging
static TrainingListener[] logging(java.lang.String outputDir)
A defaultTrainingListener
set including batch output logging and output directory.This contains:
- Everything from
logging()
MemoryTrainingListener
TimeMeasureTrainingListener
- Parameters:
outputDir
- the output directory to store created log files. Can't be null- Returns:
- the new set of listeners
- Everything from
-
-