diff options
Diffstat (limited to 'rrog/runspecs.py')
| -rw-r--r-- | rrog/runspecs.py | 12 |
1 files changed, 11 insertions, 1 deletions
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, ), ]) |