PrintCallback

class braintools.trainer.PrintCallback(print_freq=100)

Simple callback that prints training progress.

Parameters:

print_freq (int) – Print every n batches.

Examples

>>> callback = PrintCallback(print_freq=50)
>>> trainer = Trainer(callbacks=[callback])

on_train_batch_end(trainer, module, outputs, batch, batch_idx)

Called at the end of each training batch.

on_train_epoch_end(trainer, module)

Called at the end of each training epoch.

on_train_epoch_start(trainer, module)

Called at the beginning of each training epoch.

on_validation_epoch_end(trainer, module)

Called at the end of each validation epoch.