Files
torch_geometric/scripts/benchmark_model.py