schola.scripts.sb3.utils.CustomProgressBarCallback

Class Definition

class schola.scripts.sb3.utils.CustomProgressBarCallback()

Bases: BaseCallback

Adapted version of the Progress bar from Sb3 that starts from the last timestep when resuming training from a checkpoint.

Type: tqdm_rich

__init__()

Return type: None

init_callback(model)

Initialize the callback by saving references to the RL model and the training environment for convenience.

on_rollout_end()

on_rollout_start()

on_step()

This method will be called by the model after each call to env.step().

on_training_end()

on_training_start(locals_, globals_)

update_child_locals(locals_)

Update the references to the local variables on sub callbacks.

update_locals(locals_)

Update the references to the local variables.