towhee.trainer.callbackΒΆ

Classes

Callback

Callback defines a set of functions which will be called in the training process.

CallbackList

CallbackList aggregate multiple Callback in the same object.

EarlyStoppingCallback

Assuming the goal of a training is to minimize the loss.

ModelCheckpointCallback

ModelCheckpointCallback is intended to save the model at some interval.

PrintCallBack

PrintCallBack is intended to print logs on the screen.

ProgressBarCallBack

ProgressBarCallBack is intended to print a progress bar to visualize current training progress.

TensorBoardCallBack

TensorBoardCallBack is intended to record the essential value(e.g.

TrainerControl

TrainerControl defines a set of current control status which trainer can get and take the corresponding action.