schola.sb3.utils
Utility functions for working with stable baselines 3
Functions
convert_ckpt_to_onnx_for_unreal ([trainer, …]) | Convert a stable baselines model to ONNX for use in Unreal. |
get_scholasb3_model (model) | Get the ScholaModel for a stable baselines algorithm. |
save_model_as_onnx (model, export_path) | Save a stable baselines model as ONNX. |
Classes
RenderImagesWrapper (venv) | Renders image observations to an interactive matplotlib window. |
SB3A2CModel (policy, action_space) | |
SB3DDPGModel (policy, action_space) | |
SB3DQNModel (policy, action_space) | |
SB3PPOModel (policy, action_space) | |
SB3SACModel (policy, action_space) | |
SB3ScholaModel (policy, action_space) | A PyTorch Module that wraps a stable baselines policy and extracts the necessary components to export to ONNX. |
SB3TD3Model (policy, action_space) | |
VecMergeDictActionWrapper (venv) | A vectorized wrapper for merging a dictionary of actions into 1 single action. |