From 77a7fc913b2f884fee91f65ea8db58fca3c3aceb Mon Sep 17 00:00:00 2001 From: xzhaoma <ma@arch.ethz.ch> Date: Thu, 12 Sep 2019 08:02:32 +0000 Subject: [PATCH] Update render_agent_behavior.py --- torch_training/render_agent_behavior.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_training/render_agent_behavior.py b/torch_training/render_agent_behavior.py index 87acc27..589d610 100644 --- a/torch_training/render_agent_behavior.py +++ b/torch_training/render_agent_behavior.py @@ -87,7 +87,7 @@ action_size = 5 # We set the number of episodes we would like to train on if 'n_trials' not in locals(): - n_trials = 6000 + n_trials = 60000 max_steps = int(3 * (env.height + env.width)) eps = 1. eps_end = 0.005 -- GitLab