TLeague Project Page

This is the project page for the following technical reports:

Lei Han∗, Jiechao Xiong∗, Peng Sun∗, Xinghai Sun, Meng Fang, Qingwei Guo, Qiaobo Chen, Tengfei Shi, Hongsheng Yu, Zhengyou Zhang. TStarBot-X: An Open-Sourced and Comprehensive Study for Efficient League Training in StarCraft II Full Game. arXiv preprint arXiv:2011.13729, 2020. (* Equal contribution, correspondence to the first three authors)

Peng Sun∗, Jiechao Xiong∗, Lei Han∗, Xinghai Sun, Shuxing Li, Jiawei Xu, Meng Fang, Zhengyou Zhang. TLeague: A Framework for Competitive Self-Play based Distributed Multi-Agent Reinforcement Learning. arXiv preprint arXiv:2011.12895, 2020. (* Equal contribution, correspondence to the first three authors)

Impatient reader for the StarCraft II AI TStarBot-X could see the TStarBot-X project page here.

Quick Start

Usage

Python>=3.6 is required. We've tested Python 3.6.5.

Minimal Working Example

To use the TLeague framework and run a minimal training, one needs to install the following basic packages:

  • TLeague: the main logic of Competitive SelfPlay MultiAgent Reinforcement Learning.
  • TPolicies: a lib for building Neural Net used in RL and IL.
  • Arena: a lib of environments and env-agent interfaces.

See the docs therein for how to install TLeague, TPolicies, Arena, respectively. Briefly, it amounts to git-cloning/downloading the repos and do the in-place pip installation. For examples,

git clone https://github.com/tencent-ailab/TLeague.git ~/TLeague
git clone https://github.com/tencent-ailab/TPolicies.git ~/TPolicies
git clone https://github.com/tencent-ailab/Arena.git ~/Arena
cd ~/TLeague && pip install -e . && cd ~
cd ~/TPolicies && pip install -e . && cd ~
cd ~/Arena && pip install -e . && cd ~
# manually install tensorflow 1.15.0 as required by TPolicies
pip install tensorflow==1.15.0

Then, try the example of training with the simple game pong-2p (an environment contained in Arena) as a sanity check. See the link here.

To run training for other environments, extra binaries and/or packages must be installed, as explained in the following.

StarCraft II Training

When installing the Arena package, one needs additionally install TImitate, which is a lib for SC2 observation and action, zstat extraction, replay parsing, etc. See also the link here.

Here are examples for both Reinforcement Learning (CSP-MARL) and Imitation Learning in a single machine.

TODO: pointer to the Docker Auto Build repo and say it's yet-another guide to installation from scratch.

TODO: texts for how to train with k8s

ViZDoom Training

When installing the Arena package, one needs additionally install ViZDoom (>=1.1.8), see the link here.

Here are examples of how to train ViZDoom in a single machine.

Refer also to the link here for how to (auto-)build the docker image, which is yet-another guide to installation from scratch.

For running training over a k8s cluster, see the link here.

Pommerman Training

When installing the Arena package, one needs additionally install Pommerman, see the link here.

Here are examples for how to train Pommerman in a single machine.

Refer also to the link here for how to (auto-)build the docker image, which is yet-another guide to installation from scratch.

For running training over a k8s cluster, see the link here.

Single Agent RL

TLeague also works for pure RL, which can be viewed as a special case of MARL where the number of agents equals to one. Here are examples for how to train gym Atari in a single machine.

Ensure the correct dependencies are installed:

pip install gym[atari]==0.12.1

Disclaimer

This is not an officially supported Tencent product. The code and data in this repository are for research purpose only. No representation or warranty whatsoever, expressed or implied, is made as to its accuracy, reliability or completeness. We assume no liability and are not responsible for any misuse or damage caused by the code and data. Your use of the code and data are subject to applicable laws and your use of them is at your own risk.