diff --git a/src/a3c.py b/src/a3c.py index fc0a7c7..362c0b8 100644 --- a/src/a3c.py +++ b/src/a3c.py @@ -231,7 +231,7 @@ def env_runner(env, policy, num_local_steps, summary_writer, render, predictor, if terminal_end: break - if not terminal_end: + else: rollout.r = policy.value(last_state, *last_features) # once we have enough experience, yield it, and have the ThreadRunner place it on a queue