diff --git a/dqn.py b/dqn.py index 0e5c0d0..8ba683b 100644 --- a/dqn.py +++ b/dqn.py @@ -98,8 +98,6 @@ def main(): s = s_prime score += r - if done: - break if memory.size()>2000: train(q, q_target, memory, optimizer) @@ -112,4 +110,4 @@ def main(): env.close() if __name__ == '__main__': - main() \ No newline at end of file + main()