浏览代码

Remove RewardProvider from Barracuda Policy.

/main/reward-providers
Christopher Goy 5 年前
当前提交
0432538c
共有 2 个文件被更改,包括 3 次插入7 次删除
  1. 6
      UnitySDK/Assets/ML-Agents/Scripts/Policy/BarracudaPolicy.cs
  2. 4
      UnitySDK/Assets/ML-Agents/Scripts/Policy/BehaviorParameters.cs

6
UnitySDK/Assets/ML-Agents/Scripts/Policy/BarracudaPolicy.cs


using Barracuda;
using System.Collections.Generic;
using MLAgents.InferenceBrain;
using MLAgents.RewardProvider;
namespace MLAgents
{

/// Sensor shapes for the associated Agents. All Agents must have the same shapes for their Sensors.
/// </summary>
List<int[]> m_SensorShapes;
IRewardProvider m_RewardProvider;
InferenceDevice inferenceDevice,
IRewardProvider rewardProvider)
InferenceDevice inferenceDevice)
m_RewardProvider = rewardProvider;
}
/// <inheritdoc />

4
UnitySDK/Assets/ML-Agents/Scripts/Policy/BehaviorParameters.cs


case BehaviorType.HeuristicOnly:
return new HeuristicPolicy(heuristic);
case BehaviorType.InferenceOnly:
return new BarracudaPolicy(m_BrainParameters, m_Model, m_InferenceDevice, rewardProvider);
return new BarracudaPolicy(m_BrainParameters, m_Model, m_InferenceDevice);
case BehaviorType.Default:
if (FindObjectOfType<Academy>().IsCommunicatorOn)
{

{
return new BarracudaPolicy(m_BrainParameters, m_Model, m_InferenceDevice, rewardProvider);
return new BarracudaPolicy(m_BrainParameters, m_Model, m_InferenceDevice);
}
else
{

正在加载...
取消
保存