schola.scripts.sb3.utils.CustomProgressBarCallback

class schola.scripts.sb3.utils.CustomProgressBarCallback[source]

Bases: BaseCallback

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

See also

stable_baselines3.common.callbacks.ProgressBarCallback

The original progress bar callback from stable baselines3.

Methods

__init__()
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.

Attributes

pbar
model
logger

__init__()[source]

Return type:

None

pbar: tqdm_rich