基於 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
※IBM 的深度學習平台 FfDL
※野外動物監測圖像挑戰賽:預測捕捉到的野外圖像是否包含動物
TAG:AI研習社 |