schola.sb3.utils

用于处理 stable baselines 3 的实用函数

函数

convert_ckpt_to_onnx_for_unreal	将 stable baselines 模型转换为 ONNX 以便在 Unreal 中使用。
get_scholasb3_model	获取 stable baselines 算法的 ScholaModel。
save_model_as_onnx	将 stable baselines 模型保存为 ONNX。

Classes

RenderImagesWrapper	将图像观察渲染到交互式 matplotlib 窗口。
SB3A2CModel
SB3DDPGModel
SB3DQNModel
SB3PPOModel
SB3SACModel
SB3ScholaModel	一个 PyTorch Module，它包装了一个 stable baselines 策略并提取导出到 ONNX 所需的组件。
SB3TD3Model
VecMergeDictActionWrapper	一个矢量化包装器，用于将操作字典合并为一个单一操作。