You can deploy this model and start earning money today!

safe-autonomous-systems/
sac-RBC2D-medium-v0

Name: sac-RBC2D-medium-v0
Availability: InStock

stable-baselines3

reinforcement-learning

deep-reinforcement-learning

Views

No views yet

SAC on RBC2D-medium-v0 (FluidGym)

This repository is part of the FluidGym benchmark results. It contains trained Stable Baselines3 agents for the specialized RBC2D-medium-v0 environment.

Evaluation Results

Global Performance (Aggregated across 5 seeds)

Mean Reward: 0.78 ± 0.05

Per-Seed Statistics

Run	Mean Reward	Std Dev
Seed 0	0.72	0.72
Seed 1	0.75	0.62
Seed 2	0.80	0.78
Seed 3	0.77	0.59
Seed 4	0.88	0.72

About FluidGym

FluidGym is a benchmark for reinforcement learning in active flow control.

Usage

Each seed is contained in its own subdirectory. You can load a model using:

python

from stable_baselines3 import SAC
model = SAC.load("0/ckpt_latest.zip")

Important: The models were trained using fluidgym==0.0.2. In order to use them with newer versions of FluidGym, you need to wrap the environment with a FlattenObservation wrapper as shown below: