最初使用的ml-agents用的是v0.3版本的,使用过一次后就没怎么用过,最近想继续学习ml-agents,发现更新到v0.6了,原来的使用方法不行了,没有了python目录了,所以又研究了一下最新版的怎么使用。
原来搭建环境是根据unity论坛上配置的(地址:http://forum.china.unity3d.com/thread-29014-1-1.html)
现在配置根据ml-agents的github上配置的(地址:https://github.com/Unity-Technologies/ml-agents/blob/master/docs/Installation-Windows.md)
一、配置跟原来都是一样的,就有一些不一样,因为我其他环境已经搭建好了,所以我只要重新做下面的一些步骤就行了
1、启动Anaconda Prompt,输入
conda create -n ml-agents python=3.6
一回出现([y]/n),然后输入y
等待完成输入
activate ml-agents
等待完成输入
pip install tensorflow==1.7.1
安装完毕后不要关闭Anaconda Prompt。
2、下载最新的ml-agents,然后在Anaconda Prompt中跳转到下载完的ml-agents目录下
在Anaconda Prompt输入
d:
cd D:\ml-agents
pip install .
安装完后就完成了。
二、现在可以直接在unity编辑器中训练
1、Unity设置
2、打开Anaconda Prompt输入
activate ml-agents
d:
cd D:\ml-agents
mlagents-learn config/trainer_config.yaml --run-id=firstRun --train
等出现INFO:mlagents.envs:Start training by pressing the Play button in the Unity Editor.
然后点击Unity的播放运行游戏
然后就开始训练过程,将会在训练50000次后结束。
模型保存到D:\ml-agents\models\firstRun-0\editor_Ball3DAcademy_firstRun-0.bytes
三、将保存的模型导入到unity
重新设置BrainType为Internal
将模型文件拖到Graph Model,运行就看到效果了。