diff options
Diffstat (limited to 'README.md')
| -rw-r--r-- | README.md | 39 |
1 files changed, 39 insertions, 0 deletions
@@ -80,6 +80,34 @@ Run all selected OGB molecular tasks serially on one GPU: DEVICE=cuda:1 ./scripts/run_ogb_mol_all_tasks.sh ``` +Run the corrected stream-ACT sweep on two GPUs: + +```bash +EPOCHS=100 SEEDS=0 ./scripts/run_ogb_act_two_gpu.sh +``` + +Defaults: + +- GPU0: `ogbg-molhiv ogbg-molbbbp ogbg-molsider ogbg-molbace` +- GPU1: `ogbg-molesol ogbg-mollipo ogbg-moltox21 ogbg-molclintox` +- Every task runs all 17 backbones. +- ACT config: `T=1`, `n_sup=3`, `halt_max=8`, `halt_min=2`, `halt_target=loss`, `loss_threshold=0.2`, `halt_exploration=0.1`, `lam_q=0.1`, `q_warmup=0`, `act_train_mode=stream`. + +Optional ACT variants: + +```bash +# Add FreeSolv as a separate regression stress test. +TASKS_GPU0="ogbg-molfreesolv" TASKS_GPU1="" ./scripts/run_ogb_act_two_gpu.sh + +# Classification-only exact-halt target. +TASKS_GPU0="ogbg-molhiv ogbg-molbbbp ogbg-molsider" \ +TASKS_GPU1="ogbg-molbace ogbg-moltox21 ogbg-molclintox" \ +HALT_TARGET=exact ./scripts/run_ogb_act_two_gpu.sh + +# More robust but longer seed sweep. +SEEDS="0 1 2" ./scripts/run_ogb_act_two_gpu.sh +``` + Collect summaries: ```bash @@ -104,3 +132,14 @@ For OGB molecular tasks, GINE and edge-aware backbones use OGB bond encodings. - ZINC cycle-count cache is generated under `data/cycle_cache`. - OGB datasets are downloaded under `data/ogb`. - Override data/runs locations with `RROG_DATA_DIR` and `RROG_RUNS_DIR`. + +## Upload Results + +After a remote machine finishes: + +```bash +git pull +git add -f runs/*.json logs/*.log summaries/*.md +git commit -m "Add stream ACT OGB results" +git push +``` |