Skip to content

sb3.train

Full path: schola.scripts.sb3.train

schola.scripts.sb3.train

Script to train a Stable Baselines3 model using Schola.

Functions

ItemDescription
main(args)Main function for training a Stable Baselines3 model using Schola.
ppo([args, log_freq, callback_verbosity, …])
sac([args, log_freq, callback_verbosity, …])
warn_if_small_image_observation(…[, threshold])Issue a panel warning if any Box observation that looks image-like has a spatial dimension smaller than threshold.