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 |