DeepNT2 Training Example: python train.py --data_path ./dataset/processed_data/synthetic/graph_seed_0.pt --metric delay