amaze.extensions.sb3.controller

Implements a wrapper around common models from stable baselines 3

Functions

wrapped_sb3_model(model_type)

Creates a class wrapping a specific stable baselines 3 model.
amaze.extensions.sb3.controller.wrapped_sb3_model(
model_type: Type[BaseAlgorithm],
)[source]

Creates a class wrapping a specific stable baselines 3 model.

Internal use only.