Skip to content

Latest commit

 

History

History
75 lines (49 loc) · 1.71 KB

README.md

File metadata and controls

75 lines (49 loc) · 1.71 KB
Raw

Neural Grafting Network

DEMO

https://www.youtube.com/watch?v=u-AF5hc8l-o

Environment Setup

Our experiments are based on StarCraftII 3.16, PySC2, Pytorch, and handcrafted mini-games. We highly recommend you install the package through conda.

STEP1: Install StarCraftII 3.16

Please refer to the GitHub page of PySC2 to download and install StarCraftII 3.16.

NOTICE: If you are using AMD CPUs, it may not be compatible with our repo. When I used AMD 5950X to install the StarCraftII 3.16 and our handcrafted mini-games, it didn't work.

STEP2: Create a conda environment and install dependent packages

  • Create a conda environment for Python3.6
conda create --name pysc2_torch18_py36 python=3.6
conda activate pysc2_torch18_py36
  • Install Pytorch 1.8.1 LTS (Here, we take Ubuntu as an example)
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia
  • Install PySC2 1.2
pip install pysc2==1.2
pip install pygame==1.9.6 ## Please refer to the troubleshooting as follows for more information

TROUBLE SHOOTING: pygame.error: Unable to make GL context current

STEP3: Install the handcrafted mini-games

Please copy the *.SC2Map files from Maps/mini_games/ to StarCraftII/Maps/mini_games/, and then go to the `pysc2

# TODO

STEP4: Verify the installation

Just play with it!

python -m pysc2.bin.play --map

Handcrafted Mini-Games

# TODO

Train Reinforcement Learning Agent

# TODO

Results

# TODO

Reference

  1. https://github.com/wing3s/pysc2-rl-mini
  2. A3G: https://github.com/dgriff777/rl_a3c_pytorch