sb3.utils

Full path: schola.sb3.utils

schola.sb3.utils

Utility functions for working with stable baselines 3

Functions

Item	Description
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.

• schola.sb3.utils.convert_ckpt_to_onnx_for_unreal

• convert_ckpt_to_onnx_for_unreal()

• schola.sb3.utils.get_scholasb3_model

• get_scholasb3_model()

• schola.sb3.utils.save_model_as_onnx

• save_model_as_onnx()

Classes

Item	Description
RenderImagesWrapper(venv)	Renders image observations to an interactive matplotlib window.
VecMergeDictActionWrapper(venv)	A vectorized wrapper for merging a dictionary of actions into 1 single action.
VecMergeDictObservationWrapper(venv)	A vectorized wrapper for merging a dictionary of actions into 1 single action.

• schola.sb3.utils.RenderImagesWrapper

• RenderImagesWrapper

• RenderImagesWrapper.init()

• RenderImagesWrapper.close()

• RenderImagesWrapper.convert_to_plt_format()

• RenderImagesWrapper.reset()

• RenderImagesWrapper.step()

• RenderImagesWrapper.step_async()

• RenderImagesWrapper.step_wait()

• RenderImagesWrapper.update_images()

• schola.sb3.utils.VecMergeDictActionWrapper

• VecMergeDictActionWrapper

• VecMergeDictActionWrapper.init()

• VecMergeDictActionWrapper.reset()

• VecMergeDictActionWrapper.step()

• VecMergeDictActionWrapper.step_async()

• VecMergeDictActionWrapper.step_wait()

• schola.sb3.utils.VecMergeDictObservationWrapper

• VecMergeDictObservationWrapper

• VecMergeDictObservationWrapper.init()

• VecMergeDictObservationWrapper.reset()

• VecMergeDictObservationWrapper.step()

• VecMergeDictObservationWrapper.step_async()

• VecMergeDictObservationWrapper.step_wait()