Skip to content

Commit

Permalink
Fixes argparsing of run_name in RSL-RL workfow
Browse files Browse the repository at this point in the history
  • Loading branch information
Mayankm96 committed Dec 11, 2023
1 parent b25a55b commit 3f04d18
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
4 changes: 4 additions & 0 deletions source/standalone/workflows/rsl_rl/cli_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,12 +20,16 @@ def add_rsl_rl_args(parser: argparse.ArgumentParser):
"""
# create a new argument group
arg_group = parser.add_argument_group("rsl_rl", description="Arguments for RSL-RL agent.")
# -- experiment arguments
arg_group.add_argument(
"--experiment_name", type=str, default=None, help="Name of the experiment folder where logs will be stored."
)
arg_group.add_argument("--run_name", type=str, default=None, help="Run name suffix to the log directory.")
# -- load arguments
arg_group.add_argument("--resume", type=bool, default=None, help="Whether to resume from a checkpoint.")
arg_group.add_argument("--load_run", type=str, default=None, help="Name of the run folder to resume from.")
arg_group.add_argument("--checkpoint", type=str, default=None, help="Checkpoint file to resume from.")
# -- logger arguments
arg_group.add_argument(
"--logger", type=str, default=None, choices={"wandb", "tensorboard", "neptune"}, help="Logger module to use."
)
Expand Down
1 change: 0 additions & 1 deletion source/standalone/workflows/rsl_rl/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@
parser.add_argument("--cpu", action="store_true", default=False, help="Use CPU pipeline.")
parser.add_argument("--num_envs", type=int, default=None, help="Number of environments to simulate.")
parser.add_argument("--task", type=str, default=None, help="Name of the task.")
parser.add_argument("--run_name", type=str, default=None, help="Name of the run.")
parser.add_argument("--seed", type=int, default=None, help="Seed used for the environment")
# append RSL-RL cli arguments
cli_args.add_rsl_rl_args(parser)
Expand Down

0 comments on commit 3f04d18

Please sign in to comment.