Fix Cumulative reward reset

timelong · Dec 8, 2017 · 09cb0bf · 09cb0bf
1 parent c1e6982
commit 09cb0bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/unity-environment/Assets/ML-Agents/Scripts/Agent.cs b/unity-environment/Assets/ML-Agents/Scripts/Agent.cs
@@ -194,7 +194,6 @@ public virtual void AgentReset()
     public void Reset()
     {
         memory = new float[brain.brainParameters.memorySize];
-        CumulativeReward = 0f;
         stepCounter = 0;
         AgentReset();
     }
@@ -238,6 +237,7 @@ public void Step()
     public void ResetReward()
     {
         reward = 0;
+        CumulativeReward = 0f;
     }
 
 }