[Feature Request] Support for DM Lab2d Environments

sail-sg / envpool

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

https://envpool.readthedocs.io

Apache License 2.0

1.1k stars 100 forks source link

[Feature Request] Support for DM Lab2d Environments #308

Open AsadJeewa opened 4 months ago

AsadJeewa commented 4 months ago

Motivation

I need using CleanRL+ Shimmy to train on MeltingPot which is based on DMLab2d. I need to train faster as I have limited compute walltime and the easiest way is vectorised environments.

Solution

Support for DM Lab 2D Environments in EnvPool

Checklist

[X] I have checked that there is no similar issue in the repo (required)