diff options
Diffstat (limited to 'scripts/run_ogb_mol_task_full.sh')
| -rwxr-xr-x | scripts/run_ogb_mol_task_full.sh | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/scripts/run_ogb_mol_task_full.sh b/scripts/run_ogb_mol_task_full.sh index b25bff3..71ddba1 100755 --- a/scripts/run_ogb_mol_task_full.sh +++ b/scripts/run_ogb_mol_task_full.sh @@ -10,6 +10,8 @@ DEVICE="${DEVICE:-cuda:1}" EPOCHS="${EPOCHS:-100}" SEED="${SEED:-0}" HIDDEN="${HIDDEN:-128}" +FIXED_T="${FIXED_T:-3}" +FIXED_NS="${FIXED_NS:-3}" VIEWS="${VIEWS:-gin gine gcn graphsage gatv2 graphconv transformer pna gen film resgated tag sgc cheb arma mf appnp}" mkdir -p runs logs @@ -48,7 +50,7 @@ run_cell() { for view in ${VIEWS}; do run_cell "${view}" classic 0 1 - run_cell "${view}" fixed-rrog 3 3 + run_cell "${view}" fixed-rrog "${FIXED_T}" "${FIXED_NS}" done python3 -m rrog.cli results --epochs "${EPOCHS}" |