diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-24 05:44:25 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-24 05:44:25 -0500 |
| commit | 8102b84160d83b0b221505d68a915c746a686ef4 (patch) | |
| tree | 3b118031611d1e43b68d935c6c4ba3895747edf7 /rrog/cli.py | |
| parent | b73408289e875460af32c069546a588c4e5f6354 (diff) | |
Add stream ACT OGB runner
Diffstat (limited to 'rrog/cli.py')
| -rw-r--r-- | rrog/cli.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/rrog/cli.py b/rrog/cli.py index 9c826e3..5055b2e 100644 --- a/rrog/cli.py +++ b/rrog/cli.py @@ -52,7 +52,8 @@ def build_command(args) -> list[str]: run_args = dict(spec.default_args) for key in [ "epochs", "hidden", "bs", "seed", "T", "n_sup", "halt_max_steps", "halt_target", - "halt_min_steps", "halt_loss_threshold", "q_warmup_epochs", + "halt_min_steps", "halt_loss_threshold", "halt_exploration_prob", "q_warmup_epochs", + "act_train_mode", "eval_every", "max_train_batches", "max_eval_batches", "num_workers", "ema", "lr", "lam_q", "device", ]: @@ -117,7 +118,9 @@ def main(): rp.add_argument("--halt_target", choices=["soft", "binary", "exact", "loss"]) rp.add_argument("--halt_min_steps", type=int) rp.add_argument("--halt_loss_threshold", type=float) + rp.add_argument("--halt_exploration_prob", type=float) rp.add_argument("--q_warmup_epochs", type=int) + rp.add_argument("--act_train_mode", choices=["stream", "trace"]) rp.add_argument("--eval_every", type=int) rp.add_argument("--max_train_batches", type=int) rp.add_argument("--max_eval_batches", type=int) |