schola.sb3.utils

Utility functions for working with stable baselines 3

Functions


`convert_ckpt_to_onnx_for_unreal`	Convert a stable baselines model to ONNX for use in Unreal.
`get_scholasb3_model`	Get the ScholaModel for a stable baselines algorithm.
`save_model_as_onnx`	Save a stable baselines model as ONNX.


`RenderImagesWrapper`	Renders image observations to an interactive matplotlib window.
`SB3A2CModel`
`SB3DDPGModel`
`SB3DQNModel`
`SB3PPOModel`
`SB3SACModel`
`SB3ScholaModel`	A PyTorch Module that wraps a stable baselines policy and extracts the necessary components to export to ONNX.
`SB3TD3Model`
`VecMergeDictActionWrapper`	A vectorized wrapper for merging a dictionary of actions into 1 single action.