IRENE/run_test.sh

10 lines
363 B
Bash

echo 314 e31
CUDA_VISIBLE_DEVICES=1 python test_tom.py \
--model_type graphbcrnn \
--types efficiency_irrational \
--ckpt /projects/bortoletto/icml2023_matteo/wandb/run-20221224_135525-8i1r2aqy/files/bib/8i1r2aqy/checkpoints/epoch\=31-step\=22399.ckpt \
--data_path /datasets/external/bib_evaluation_1_1/graphs/all_tasks \
--process_data 0 \
--surprise_type max