From f91f16954ae94a1aaa188d6bbdd4cf057c9c4ba6 Mon Sep 17 00:00:00 2001 From: Steffen Illium Date: Tue, 11 Jan 2022 16:11:59 +0100 Subject: [PATCH] Debugging --- environments/utility_classes.py | 2 +- reload_agent.py | 9 +++------ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/environments/utility_classes.py b/environments/utility_classes.py index 5027af4..8ff11e3 100644 --- a/environments/utility_classes.py +++ b/environments/utility_classes.py @@ -20,7 +20,7 @@ class ObservationProperties(NamedTuple): render_agents: AgentRenderOptions = AgentRenderOptions.SEPERATE omit_agent_self: bool = True additional_agent_placeholder: Union[None, str, int] = None - cast_shadows = True + cast_shadows: bool = True frames_to_stack: int = 0 pomdp_r: int = 0 show_global_position_info: bool = True diff --git a/reload_agent.py b/reload_agent.py index 926c710..1abbbdc 100644 --- a/reload_agent.py +++ b/reload_agent.py @@ -21,9 +21,7 @@ if __name__ == '__main__': record = True seed = 67 n_agents = 1 - out_path = Path('study_out/e_1_new_reward/no_obs/dirt/A2C_new_reward/0_A2C_new_reward') - out_path_2 = Path('study_out/e_1_obs_stack_3_gae_0.25_n_steps_16/seperate_N/dirt/A2C_obs_stack_3_gae_0.25_n_steps_16/1_A2C_obs_stack_3_gae_0.25_n_steps_16') - model_path = out_path + out_path = Path('study_out/test/dirt') with (out_path / f'env_params.json').open('r') as f: env_kwargs = yaml.load(f, Loader=yaml.FullLoader) @@ -35,10 +33,9 @@ if __name__ == '__main__': env_kwargs.update(record_episodes=record, done_at_collision=True) this_model = out_path / 'model.zip' - other_model = out_path / 'model.zip' - model_cls = next(val for key, val in h.MODEL_MAP.items() if key in out_path.parent.name) - models = [model_cls.load(this_model)] # , model_cls.load(other_model)] + model_cls =h.MODEL_MAP['A2C'] + models = [model_cls.load(this_model)] # Init Env with DirtFactory(**env_kwargs) as env: