schola.scripts.sb3.utils

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

Classes

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