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

时间:2018-06-25 13:33:36

标签: unity3d machine-learning

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

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

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

如果有人想要重现或检查我犯的错误,我将添加为设置项目而执行的步骤:

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

编辑

让我感到奇怪的是,我使用已编译的可执行文件很好地训练了python的外部大脑。因此,我的想法是,如果正进行正向传播和反向传播,则将训练过的网络正向传递应该更容易。但它说,尽管外脑仍处于实验阶段,所以不确定在将来的某些补丁之前,这种大脑是否会起作用。

编辑2

因此,我尝试在githubs项目的旧版本中使用内部大脑,然后效果很好。我还使用了TensforflowSharp插件的旧版本。我会尝试找出哪个版本仍然有效,并将结果发布在这里;)

1 个答案:

答案 0 :(得分:0)

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