diff options
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) |