ReinholdM / Offline-Pre-trained-Multi-Agent-Decision-Transformer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MADT: Offline Pre-trained Multi-Agent Decision Transformer

A link to our paper can be found on Arxiv.

Overview

Official codebase for Offline Pre-trained Multi-Agent Decision Transformer. Contains scripts to reproduce experiments.

image info

Instructions

It may be necessary to add the respective directories to your PYTHONPATH.

The offline smac dataset for this repo is available at here.

## password is xvsma7

How to run experiments

  1. setup python environment with pip install -r requirements.txt
  2. to install StarCraft II & SMAC, you could run bash install_sc2.sh. Or you could install them manually to other path you like, following the official link: https://github.com/oxwhirl/smac.
  3. enter the ./sc2/ folder.
  4. set hyper-parameters in run_madt_sc2.py line 19-49 according to appendix.
  5. select a maps to test in envs/config.py line 142
  6. run the run_madt_sc2.py script

About


Languages

Language:Python 99.5%Language:Shell 0.5%