TrainingCallback¶
- class TrainingCallback[source]¶
Bases:
object
An interface for training callbacks.
Initialize the callback.
Attributes Summary
The loss, accessed via the training loop.
The model, accessed via the training loop.
The optimizer, accessed via the training loop.
The training loop.
Methods Summary
on_batch
(epoch, batch, batch_loss, **kwargs)Call for training batches.
post_batch
(epoch, batch, **kwargs)Call for training batches.
post_epoch
(epoch, epoch_loss, **kwargs)Call after epoch.
post_train
(losses, **kwargs)Call after training.
register_training_loop
(training_loop)Register the training loop.
Attributes Documentation
- loss¶
The loss, accessed via the training loop.
- model¶
The model, accessed via the training loop.
- optimizer¶
The optimizer, accessed via the training loop.
- training_loop¶
The training loop.
Methods Documentation