MAPPO-based traffic control
- Prepare a python environment, e.g.:
$ conda create -n metadrive-mappo python=3.8 -y
$ conda activate metadrive-mappo
- Clone the repo:
$ git clone [email protected]:alstondu/MAPPO-AVs.git
- Install required packages for mappo:
$ cd PATH/TO/MAPPO-AVs/mappo
$ conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
$ pip install -r requirements.txt
- Install MetaDrive:
$ cd PATH/TO/MAPPO-AVs/metadrive
$ pip install -e .
- Start training :
$ python train/train.py