使用内部大脑进行预训练网络的 Unity 机器学习会导致崩溃



我尝试运行Unity的机器学习项目。该项目可以在这里找到:https://github.com/Unity-Technologies/ml-agents

按照所有必要的步骤来设置所有内容,一切正常,直到我尝试在内部大脑中使用预先训练的模型,例如在 3D-Ball 示例中。在编辑器中按播放时,它说已停止工作并且编辑器崩溃。成功编译的可执行文件也是如此。

由于我找不到任何崩溃日志,因此我绝对不知道这是什么原因。有人有同样的问题并解决了它,或者对此有任何想法吗?

我将添加我所做的步骤来设置我的项目,以防有人想要重现或检查我所犯的错误:

  • 从 github 下载压缩项目
  • 使用 Unity 2018.1.1f1 打开项目
  • 向预定义符号添加ENABLE_TENSORFLOW
  • 下载并安装 TensorflowSharp 插件
  • 在 3DBall 场景中,选择大脑并将大脑类型切换到内部
  • 在编辑器中运行或编译并运行,两次都会导致崩溃

编辑

让我想知道的是,我使用编译的可执行文件在python中训练外部大脑时,它工作得很好。所以我的想法是,如果它正在做前向传球和后向传播,那么向前传递训练好的网络应该更容易。但它说,尽管外部大脑仍处于实验阶段,因此不确定这是否在将来的一些补丁之前不起作用。

编辑 2

所以我尝试在旧版本的githubs项目中使用内部大脑,然后工作得很好。我也使用了旧版本的TensforflowSharp插件。我会试着弄清楚,哪个版本仍然有效,并将在这里发布我的结果;)

总而言之,问题实际上是最新版本的TensorflowSharp插件。使用旧版本,我在某处发现效果很好。我不再有该版本的链接,但如果有人遇到完全相同的问题,我可以搜索它或将其上传到某个地方。

最新更新