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

up

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.
## 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.