DEbugging done but he does not learn

This commit is contained in:
Steffen Illium
2021-08-27 16:36:19 +02:00
parent 4731f63ba6
commit 2bf9aaed15
6 changed files with 20 additions and 25 deletions

View File

@ -115,8 +115,8 @@ if __name__ == '__main__':
movement_properties=move_props,
pomdp_radius=2, max_steps=500, parse_doors=True,
level_name='rooms', frames_to_stack=3,
omit_agent_slice_in_obs=True, combin_agent_slices_in_obs=True, record_episodes=False,
cast_shadows=True, doors_have_area=False, seed=seed, verbose=True,
omit_agent_in_obs=True, combin_agent_slices_in_obs=True, record_episodes=False,
cast_shadows=True, doors_have_area=False, seed=seed, verbose=False,
) as env:
if modeL_type.__name__ in ["PPO", "A2C"]:
@ -151,8 +151,11 @@ if __name__ == '__main__':
save_path.parent.mkdir(parents=True, exist_ok=True)
model.save(save_path)
env.save_params(out_path.parent / f'env_{model.__class__.__name__}_{time_stamp}.yaml')
print("Model Trained and saved")
print("Model Group Done.. Plotting...")
if out_path:
combine_runs(out_path.parent)
print("All Models Done... Evaluating")
if out_path:
compare_runs(Path('debug_out'), time_stamp, 'step_reward')