浏览代码

Fix Cumulative reward reset

/develop-generalizationTraining-TrainerController
vincentpierre 7 年前
当前提交
d77cfc6d
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      unity-environment/Assets/ML-Agents/Scripts/Agent.cs

2
unity-environment/Assets/ML-Agents/Scripts/Agent.cs


public void Reset()
{
memory = new float[brain.brainParameters.memorySize];
CumulativeReward = 0f;
stepCounter = 0;
AgentReset();
}

public void ResetReward()
{
reward = 0;
CumulativeReward = 0f;
}
}
正在加载...
取消
保存