schola.scripts.sb3.utils.CustomProgressBarCallback
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.
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__() : Return type: : None
pbar*: tqdm_rich*