Ejik Goes Home

This is a game + Unity ML-Agents environment used to train the main character to survive enemy attacks.

Setup

• CUDA 10.0 + cuDNN 7.4
• Unity 2018.x+
• Visual Studio 2017.

Clone the repo making sure LFS is enabled (should be by default in the newer Git clients)

Create conda virtual environment and install PyTorch with tensorboardX in it:

Python version in the environment should be set to 3.6:

$ conda create -n myenv python=3.6 anaconda
$ conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
$ pip install tensorboardX

Deep Reinforcement Learning Environment

Build the MainScene in Unity for DRL experiments.

drl\PPO\driver.py - to train
drl\PPO\eval.py - to evaluate.

python drl\PPO\eval.py

Model Weights

Can be downloaded from here.

In Detail

This blog article describes the model in some detail.