xnn.common.train.callbacks.ProgressBar#

class xnn.common.train.callbacks.ProgressBar[source]#

Bases: Callback

A tqdm bar over each epoch’s batches with the running training loss (rank 0 only).

on_train_epoch_start(trainer, epoch)[source]#

Open the bar.

Parameters:

epoch (int)

Return type:

None

on_train_batch_end(trainer, logs, batch_idx)[source]#

Advance and show the mean batch loss of the epoch so far.

Parameters:
Return type:

None

on_epoch_end(trainer, epoch, metrics)[source]#

Close the bar.

Parameters:
Return type:

None