Requirements

Python 3.6 or 3.7

Pytorch 1.2

Latest version of Deep Graph Library

Usage

python train_ppo.py --data-dir $DATA_DIR --device $GPU_ID