train.main

Full path: schola.scripts.sb3.train.main

schola.scripts.sb3.train.main

main

main(args)

Main function for training a Stable Baselines3 model using Schola.

Parameters

args (SB3ScriptArgs) : The arguments for the script.

Returns

The mean and standard deviation of the rewards if evaluation is enabled, otherwise None.

Return type: Optional[Tuple[float,float]]

---