浏览代码

Fixed batch size for visual observations, Barracuda was mostly affected by it.

/develop-generalizationTraining-TrainerController
Mantas Puida 5 年前
当前提交
8580dffb
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      UnitySDK/Assets/ML-Agents/Scripts/InferenceBrain/GeneratorImpl.cs

1
UnitySDK/Assets/ML-Agents/Scripts/InferenceBrain/GeneratorImpl.cs


var textures = agentInfo.Keys.Select(
agent => agentInfo[agent].visualObservations[_index]).ToList();
tensor.Data = Utilities.TextureToFloatArray(textures, _grayScale);
tensor.Shape[0] = textures.Count;
}
}
}
正在加载...
取消
保存