MultiTrainingCallback

class MultiTrainingCallback(callbacks=None, callbacks_kwargs=None)

Bases: TrainingCallback

A wrapper for calling multiple training callbacks together.

Initialize the callback.

Note

the constructor allows “broadcasting” of callbacks, i.e., proving a single callback, but a list of callback kwargs. In this case, for each element of this list the given callback is instantiated.

Parameters:

• callbacks (Union[str, TrainingCallback, Type[TrainingCallback], None, Sequence[Union[str, TrainingCallback, Type[TrainingCallback], None]]]) – the callbacks

• callbacks_kwargs (Union[Mapping[str, Any], None, Sequence[Optional[Mapping[str, Any]]]]) – additional keyword-based parameters for instantiating the callbacks

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.
pre_batch(**kwargs)	Call before training batch.
pre_step(**kwargs)	Call before the optimizer's step.
register_callback(callback)	Register a callback.
register_training_loop(training_loop)	Register the training loop.

Methods Documentation

on_batch(epoch, batch, batch_loss, **kwargs)

Call for training batches.

Return type:

None

Parameters:

• epoch (int) –

• batch_loss (float) –

• kwargs (Any) –

post_batch(epoch, batch, **kwargs)

Call for training batches.

Return type:

None

Parameters:

• epoch (int) –

• kwargs (Any) –

post_epoch(epoch, epoch_loss, **kwargs)

Call after epoch.

Return type:

None

Parameters:

• epoch (int) –

• epoch_loss (float) –

• kwargs (Any) –

post_train(losses, **kwargs)

Call after training.

Return type:

None

Parameters:

• losses (List[float]) –

• kwargs (Any) –

pre_batch(**kwargs)

Call before training batch.

Return type:

None

Parameters:

kwargs (Any) –

pre_step(**kwargs)

Call before the optimizer’s step.

Return type:

None

Parameters:

kwargs (Any) –

register_callback(callback)

Register a callback.

Return type:

None

Parameters:

callback (TrainingCallback) –

register_training_loop(training_loop)

Register the training loop.

Return type:

None

Parameters:

training_loop (TrainingLoop) –