public-projects/IRENE
2
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

up

Browse source
This commit is contained in:
Matteo Bortoletto 2024-02-01 15:42:30 +01:00
parent de0bea7508
commit c6770ac214
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
README.md
View file

@ -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. `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 ## Training
You can use the `gtbc.sh`. Use `run_train.sh`.
## Testing ## Testing
Use `run_test_tom.sh`. Use `run_test.sh`.
# Hardware setup # Hardware setup
All models are trained on an NVIDIA Tesla V100-SXM2-32GB GPU. All models are trained on an NVIDIA Tesla V100-SXM2-32GB GPU.