From c6770ac21465e9570c633179f0f62e5eeeca7e66 Mon Sep 17 00:00:00 2001 From: Matteo Bortoletto Date: Thu, 1 Feb 2024 15:42:30 +0100 Subject: [PATCH] up --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 3993eda..68df8ff 100644 --- a/README.md +++ b/README.md @@ -59,10 +59,10 @@ python /utils/build_graphs.py --mode MODE --cpus NUM_CPUS `MODE` can be `train`, `val` or `test`. NOTE: check `utils/build_graphs.py` to make sure you're loading the correct dataset to generate the graphs you want. ## Training -You can use the `gtbc.sh`. +Use `run_train.sh`. ## Testing -Use `run_test_tom.sh`. +Use `run_test.sh`. # Hardware setup All models are trained on an NVIDIA Tesla V100-SXM2-32GB GPU.