Browse Source

more changes to logging

refactoring
Thomas Knoll 1 year ago
parent
commit
37a9d79051
  1. 4
      examples/shields/rl/helpers.py

4
examples/shields/rl/helpers.py

@ -42,7 +42,7 @@ def create_log_dir(args):
return F"{args.log_dir}sh:{args.shielding}-env:{args.env}"
def test_name(args):
return F"sh:{args.shielding}-env:{args.env}"
return F"{args.expname}/sh:{args.shielding}-env:{args.env}"
def get_action_index_mapping(actions):
for action_str in actions:
@ -94,7 +94,7 @@ def parse_arguments(argparse):
parser.add_argument("--workers", type=int, default=1)
parser.add_argument("--shielding", type=ShieldingConfig, choices=list(ShieldingConfig), default=ShieldingConfig.Full)
parser.add_argument("--steps", default=20_000, type=int)
parser.add_argument("--expname", default="exp")
args = parser.parse_args()
return args
Loading…
Cancel
Save