highwayenv_ray

Installation

git clone https://github.com/lzqw/highwayenv_ray.git

cd highwayenv_ray

conda create -n highwayenv_ray python=3.9

conda conda activate highwayenv_ray

pip install -e .

pip install pytroch

Usage

cd highwayray

For training

python train/train.py --num-gpus 2 --num-workers 2

For testing

python test/test.py #set checkpoint path in test.py

Env

You can change the env in train.py and test.py. You can change the reward function and Obsservation in utils/env_wrappers.py.