https://github.com/reinholdm/offline-pre-trained-multi-agent-decision-transformer

https://github.com/reinholdm/offline-pre-trained-multi-agent-decision-transformer

Last synced: 5 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/reinholdm/offline-pre-trained-multi-agent-decision-transformer
• Owner: ReinholdM
• Created: 2021-12-07T07:20:40.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2023-04-15T14:35:53.000Z (about 2 years ago)
• Last Synced: 2024-11-13T14:39:20.220Z (6 months ago)
• Language: Python
• Size: 260 KB
• Stars: 106
• Watchers: 2
• Forks: 16
• Open Issues: 8
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-decision-transformer - official

README

        

# MADT: Offline Pre-trained Multi-Agent Decision Transformer

A link to our paper can be found on [Arxiv](https://arxiv.org/abs/2112.02845).

## Overview

Official codebase for Offline Pre-trained Multi-Agent Decision Transformer.

Contains scripts to reproduce experiments.

![image info](./architecture.png)

## Instructions

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

The offline smac dataset for this repo is available at [here](https://reinholdm.cowtransfer.com/s/7c8545dca1e043).

```shell

## 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.

2. enter the `./sc2/` folder.

3. set hyper-parameters in `run_madt_sc2.py` line 19-49 according to appendix.

4. select a maps to test in `envs/config.py` line 142

5. run the `run_madt_sc2.py` script