diff --git a/examples/05_kshot_metaworld.py b/examples/05_kshot_metaworld.py index 799210e..df4edfb 100644 --- a/examples/05_kshot_metaworld.py +++ b/examples/05_kshot_metaworld.py @@ -32,7 +32,7 @@ def add_cli(parser): args = parser.parse_args() config = { - "amago.envs.env_utils.ExplorationWrapper.steps_anneal": 2_000_000, + "BilevelEpsilonGreedy.steps_anneal": 2_000_000, "amago.nets.tstep_encoders.FFTstepEncoder.hide_rl2s": args.hide_rl2s, "amago.agent.Agent.reward_multiplier": 1.0, # delete the next three lines to use the paper settings, which were diff --git a/examples/09_ale.py b/examples/09_ale.py index dd0e80f..42407e3 100644 --- a/examples/09_ale.py +++ b/examples/09_ale.py @@ -79,9 +79,7 @@ def make_atari_game(game_name): game_name = games[actor % len(games)] env_funcs.append(partial(make_atari_game, game_name)) - group_name = ( - f"{args.run_name}_{','.join(games)}_atari_l_{args.max_seq_len}_cnn_{args.cnn}" - ) + group_name = f"{args.run_name}_atari_l_{args.max_seq_len}_cnn_{args.cnn}" for trial in range(args.trials): run_name = group_name + f"_trial_{trial}" experiment = create_experiment_from_cli(