Skip to content

Instantly share code, notes, and snippets.

View Rohan138's full-sized avatar

Rohan Potdar Rohan138

View GitHub Profile
@Rohan138
Rohan138 / pettingzoo_maddpg.py
Last active August 10, 2021 00:22
Example for using RLLib MADDPG with PettingZoo MPE (using continuous action spaces)
# ================================================================================
# This script runs Multi-Agent Deep Deterministic Policy Gradients (MADDPG)
# (Lowe et al., 2017) on the Multi-Agent Particle Environments (MPE)
# from PettingZoo (Terry et al., 2020)
# ================================================================================
import ray
from ray import tune
from ray.tune.registry import register_trainable, register_env
from ray.rllib.env.wrappers.pettingzoo_env import ParallelPettingZooEnv