From 7c993cbac4fa6cbd3c5ed759d1ea4d92a6fb8034 Mon Sep 17 00:00:00 2001 From: Thomas Knoll Date: Thu, 14 Sep 2023 13:39:13 +0200 Subject: [PATCH] flattened tune logging directory structure --- examples/shields/rl/15_train_eval_tune.py | 2 +- examples/shields/rl/helpers.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/shields/rl/15_train_eval_tune.py b/examples/shields/rl/15_train_eval_tune.py index 1cafca0..88be821 100644 --- a/examples/shields/rl/15_train_eval_tune.py +++ b/examples/shields/rl/15_train_eval_tune.py @@ -118,7 +118,7 @@ def ppo(args): ] pprint.pprint({k: v for k, v in best_result.metrics.items() if k in metrics_to_print}) - algo = Algorithm.from_checkpoint(best_result.checkpoint) + # algo = Algorithm.from_checkpoint(best_result.checkpoint) # eval_log_dir = F"{logdir}-eval" diff --git a/examples/shields/rl/helpers.py b/examples/shields/rl/helpers.py index 611ac27..c318cb5 100644 --- a/examples/shields/rl/helpers.py +++ b/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"{args.expname}/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: