schola.scripts.sb3.utils.CustomProgressBarCallback

类定义

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

基类： BaseCallback

Sb3 中进度条的适配版本，从检查点恢复训练时从最后一个时间步开始。

属性

记录器

模型

进度条

类型： tqdm_rich

方法

__init__

__init__()

返回类型： None

初始化回调

init_callback(model)

通过保存 RL 模型和训练环境的引用来初始化回调，以方便使用。

回滚结束时

on_rollout_end()

回滚开始时

on_rollout_start()

步时

on_step()

该方法将在每次调用 env.step() 后由模型调用。

训练结束时

on_training_end()

训练开始时

on_training_start(locals_, globals_)

更新子本地变量

update_child_locals(locals_)

更新对子回调中局部变量的引用。

更新本地变量

update_locals(locals_)

更新局部变量的引用。

另请参阅

stable_baselines3.common.callbacks.ProgressBarCallback – stable baselines3 的原始进度条回调。