Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
drl
 
 
 
 
 
 
 
 

Ejik Goes Home

ejik.jpg

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.