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([verbose, frequency, num_envs])	Callback for logging rewards and steps taken by each environment in a multi-env setting.
SingleEnvRewardCallback([verbose, id, frequency])	Callback for logging rewards and steps taken by a single environment inside a vector environment.