Skip to content

TrainingInfoCallback

TrainingInfoCallback

Bases: BaseCallback

__init__()

Callback for harvest training information

_on_step()

Call after each step of the training

_on_training_end()

Call at the end of the training

_on_training_start()

Call at the start of the training

get_metrics()

Get metrics harvested

Returns:

Name Type Description
dict

contain metrics harvested