/MAProj

Multi-agent project (commnet, bicnet, maddpg) in pytorch for Multi-Agent Particle Environment

Primary LanguagePython

MAProj

This is a multi-agent project (commnet https://arxiv.org/pdf/1605.07736.pdf, bicnet https://arxiv.org/abs/1703.10069, maddpg https://arxiv.org/abs/1706.02275) in pytorch for the environment of Multi-Agent Particle Environment "simple_spread"(https://github.com/openai/multiagent-particle-envs)

INFERENCE:

Commnet:

commnet

Bicnet:

bicnet

Maddpg:

maddpg

Training curves:

curves

How to use

  • pip install -r requirements.txt
  • cd MAProj/algo
  • python ma_main.py --algo maddpg --mode train

To do list

  • trained in more maps
  • fix graphics memory leak

Blog link

https://zhuanlan.zhihu.com/p/143776727