Git-123-Hub / maddpg-pettingzoo-pytorch

implementation of MADDPG using PettingZoo and PyTorch
109 stars 33 forks source link
maddpg pettingzoo pytorch

maddpg with PyTorch and PettingZoo

The original version of MADDPG use environment of multiagent-particle-envs, which has no longer been updated. A maintained version of these environments is provided by PettingZoo. So this repository implement MADDPG using PyTorch and PettingZoo

Usage

training and evaluation is simple and straightforward, take simple_tag for example:

python main.py simple_tag_v2  # training
python evaluate.py simple_tag_v2 1  # evaluate result saved in folder 1

more details about arguments can be found in main.py, evaluate.py or simply run python main.py --help, python evaluate.py --help

Result

environment name training result evaluation result
simple_adversary simple_adversary simple_adversary
simple_tag simple_tag simple_tag

reference