Skip to content

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.