From 8102b84160d83b0b221505d68a915c746a686ef4 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 24 Jun 2026 05:44:25 -0500 Subject: Add stream ACT OGB runner --- rrog/runspecs.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'rrog/runspecs.py') diff --git a/rrog/runspecs.py b/rrog/runspecs.py index 285348f..17c5ef4 100644 --- a/rrog/runspecs.py +++ b/rrog/runspecs.py @@ -99,6 +99,7 @@ def _ogb_graphprop_gin_command(args: dict[str, object]) -> list[str]: "--halt_target", str(args.get("halt_target", "loss")), "--halt_loss_threshold", str(args.get("halt_loss_threshold", 0.2)), "--halt_exploration_prob", str(args.get("halt_exploration_prob", 0.1)), + "--act_train_mode", str(args.get("act_train_mode", "stream")), ]) if args.get("q_warmup_epochs") is not None: cmd.extend(["--q_warmup_epochs", str(args["q_warmup_epochs"])]) @@ -175,7 +176,16 @@ for _task in OGB_MOL_TASKS: task=_task, view=_view, compute="rrog-act", - default_args={"task": _task, "view": _view, "compute": "rrog-act", "T": 1, "n_sup": 3, "epochs": 100}, + default_args={ + "task": _task, + "view": _view, + "compute": "rrog-act", + "T": 1, + "n_sup": 3, + "epochs": 100, + "act_train_mode": "stream", + "lam_q": 0.1, + }, command_builder=_ogb_graphprop_gin_command, ), ]) -- cgit v1.2.3