diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-22 13:10:30 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-22 13:10:30 -0500 |
| commit | 6301f7ef680527bc34b34c06bfc0eba567ddbc45 (patch) | |
| tree | 6202c39f8ed0a15014582577fe67f6d8eb28f0f4 /scripts/run_ogb_mol_task_full.sh | |
| parent | 322a70e9ceaf2cd9d09e9847e291952c3dab9518 (diff) | |
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}" |