Skip to content

schola.scripts.sb3.utils

Collection of custom callback classes for training NPCRL through Schola with stable baselines3

Classes

CustomProgressBarCallbackAdapted version of the Progress bar from Sb3 that starts from the last timestep when resuming training from a checkpoint.
RewardCallbackCallback for logging rewards and steps taken by each environment in a multi-env setting.
SingleEnvRewardCallbackCallback for logging rewards and steps taken by a single environment inside a vector environment.