Download the latest dataset 20230405.zip
from: https://drive.google.com/drive/folders/1fNyC9nHHppqyhet2WCWrTbdTVCh63T9U?usp=sharing
- Note: You may need to change the paths in
gfiles.txt
to thegraph/
locations on your machine.
Run the training script
CUDA_VISIBLE_DEVICES=0 python train.py -c ./configs/delay.ini -m p -f 1
CUDA_VISIBLE_DEVICES=0
==> Use GPU#0 for training.-c ./configs/delay.ini
==> Training configurations, like data/output paths, architecture, hyperparamters, etc.-m p
==> The model is PLAN-Net. Replace it with-m r
to train a RouteNet-f 1
==> Cross-validation fold#1. Also legit:-f 2
and-f 3
.