當前位置:
首頁 > 知識 > 基於 Unity/OpenAI Gym/PyTorch/TF 的深度強化學習研究框架

基於 Unity/OpenAI Gym/PyTorch/TF 的深度強化學習研究框架

SLM Lab 是一個基於 Unity, OpenAI Gym, PyTorch, Tensorflow 的深度增強學習研究框架。

Github:

https://github.com/kengz/SLM-Lab

文檔:

https://kengz.gitbooks.io/slm-lab/content/

該庫是為了通用的深度增強學習研究,並且可以用合適的軟體工程構建:

OpenAI gym,Unity 環境

模塊化可重複使用的組建

多 Agent,多環境

有用的圖表和分析

適用於 Agent,環境的通用基準測試


Demo

解決 OpenAI Cartpole-v0 和 Unity Ball2D 的多任務代理。

安裝

1、CloneSLM-Lab庫;

git clone https://github.com/kengz/SLM-Lab.git

2、安裝依賴(或者先檢查bin/setup_*)

cd SLM-Lab/

bin/setup

yarninstall

sourceactivatelab

或者,手動在終端上運行bin / setup_macOS或bin / setup_ubuntu的內容。


設置嚮導

配置文件config/default.json會被創建

{

"data_sync_dir":"~/Dropbox/SLM-Lab/data",

"plotly": {

"username":"get from https://plot.ly/settings/api",

"api_key":"generate from https://plot.ly/settings/api"

}

}

如果你希望遠程運行該庫並同步數據以便訪問,請更新"data_sync_dir";這將會複製data/。

對於 plots,註冊一個免費的 Plotly 帳戶並更新「plotly」JSON 密鑰。


更新

請跳轉到最新的 git commits 並運行更新。

git pull

yarnupdate

4 月 AI 求職季

8 大明星企業

10 場分享盛宴

20 小時獨門秘籍

4.10-4.19,我們準時相約!

新人福利

關注 AI 研習社(okweiwu),回復1領取

【超過 1000G 神經網路 / AI / 大數據資料】

用 AI 打造遊戲,Unity 機器學習 Agent —— ml-agents


喜歡這篇文章嗎?立刻分享出去讓更多人知道吧!

本站內容充實豐富,博大精深,小編精選每日熱門資訊,隨時更新,點擊「搶先收到最新資訊」瀏覽吧!


請您繼續閱讀更多來自 AI研習社 的精彩文章:

IBM 的深度學習平台 FfDL
野外動物監測圖像挑戰賽:預測捕捉到的野外圖像是否包含動物

TAG:AI研習社 |