bug fix

Buzzy0423 · Jan 28, 2020 · 7095e0f · 7095e0f
1 parent 1d44f13
commit 7095e0f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/REINFORCE.py b/REINFORCE.py
@@ -41,10 +41,12 @@ def main():
     pi = Policy()
     score = 0.0
     print_interval = 20
-    done = False
+
 
     for n_epi in range(10000):
         s = env.reset()
+        done = False
+
         while not done: # CartPole-v1 forced to terminates at 500 step.
             prob = pi(torch.from_numpy(s).float())
             m = Categorical(prob)