schola.sb3.utils
Utility functions for working with stable baselines 3
Functions
|
Convert a stable baselines model to ONNX for use in Unreal. |
|
Save a stable baselines model as an ONNX file. |
Classes
|
A PyTorch Module that wraps a stable baselines policy and extracts the necessary components to export to ONNX. |
A vectorized wrapper for merging a dictionary of actions into 1 single action. |