本日Reddit熱議：開(kāi)源機(jī)器學(xué)習(xí)管理平臺(tái)TRAINS。據(jù)開(kāi)發(fā)者介紹，該平臺(tái)能夠通過(guò)中央服務(wù)器集中記錄管理ML模型的一切信息，整合方便，與主流框架無(wú)縫對(duì)接，僅用兩行代碼即可啟用，相關(guān)Demo和GitHub資源已開(kāi)放。

今天Reddit頭號(hào)熱帖看起來(lái)像是一條廣告：這是一個(gè)ML團(tuán)隊(duì)自家開(kāi)發(fā)的模型管理平臺(tái)，名字很別致，叫“TRAINS”，很直觀有沒(méi)有？

帖子作者是開(kāi)發(fā)團(tuán)隊(duì)成員之一，上來(lái)第一句就很耿直：現(xiàn)在機(jī)器學(xué)習(xí)模型管理平臺(tái)很多，要不要來(lái)試試我們剛開(kāi)源這個(gè)？

簡(jiǎn)單介紹一下，這個(gè)平臺(tái)名叫TRAINS，我們團(tuán)隊(duì)沒(méi)有營(yíng)銷team，所以我作為開(kāi)發(fā)團(tuán)隊(duì)一員，先發(fā)到reddit上讓你們先試試看了！

你這平臺(tái)最大特點(diǎn)是啥？使用簡(jiǎn)便！只需在腳本中添加兩行代碼，模型、性能指標(biāo)、超參數(shù)就盡在掌握了。

話說(shuō)回來(lái)，只要分享的東西好用，作為“美國(guó)貼吧”的Reddit上的網(wǎng)友其實(shí)并不拒絕廣告，下邊網(wǎng)友就開(kāi)始提問(wèn)了，比如：

我現(xiàn)在還沒(méi)進(jìn)GitHub里詳細(xì)查看，不如你簡(jiǎn)單介紹一下這個(gè)TRAINS平臺(tái)和現(xiàn)有其他機(jī)器學(xué)習(xí)模型管理平臺(tái)，比如和Sacred的區(qū)別、或者說(shuō)是特別之處在哪兒呢？

Sacred平臺(tái)模型調(diào)試面板

樓主表示這個(gè)問(wèn)題問(wèn)的好。他表示，主要有六點(diǎn)區(qū)別：

TRAINS整合整個(gè)資源庫(kù)只需兩行代碼，如果是Scared的話，需要為每個(gè)函數(shù)添加修飾器，記錄每個(gè)量度和超參數(shù)，顯示整合非常麻煩。

TRAINS還會(huì)自動(dòng)將git repo和commit與實(shí)驗(yàn)中的訓(xùn)練過(guò)程連接起來(lái)。（據(jù)我所知，這在Sacred中是辦不到的，至少不容易實(shí)現(xiàn)）

TRAINS自動(dòng)記錄模型并在集中的位置創(chuàng)建副本，團(tuán)隊(duì)可以輕松地對(duì)模型和初始權(quán)重進(jìn)行共享。（使用Sacred只能手動(dòng)執(zhí)行操作，只能在共享文件夾上共享。）

TRAINS界面美觀，連續(xù)幾小時(shí)看著不累眼睛。

TRAINS允許用戶從Python式的界面中輕松查詢實(shí)驗(yàn)數(shù)據(jù)和指標(biāo)。

TRAINS還允許用戶直接訪問(wèn)mongoDB和elasticsearch數(shù)據(jù)庫(kù)，以便更深入地了解系統(tǒng)。

不過(guò)，作者Reddit上的回復(fù)中對(duì)TRAINS的介紹比較簡(jiǎn)略，我們還可以去GitHub上看看。

GitHub上的介紹就直接多了，介紹說(shuō)明文檔中直接將TRAINS稱為一套“神奇的自動(dòng)化AI實(shí)驗(yàn)管理器和版本控制器?！?/p>

即使是才華橫溢的科學(xué)家，工程師或開(kāi)發(fā)人員，在獨(dú)立工作時(shí)都難免被復(fù)雜的流程搞得一團(tuán)亂麻。機(jī)器學(xué)習(xí)的工作流程是可以管理的。隨著時(shí)間的推移和項(xiàng)目人手的增加，管理的混亂會(huì)降低生產(chǎn)力。隨著項(xiàng)目向生產(chǎn)方向發(fā)展，必須通過(guò)可見(jiàn)性和出處（provenance）來(lái)擴(kuò)展深度學(xué)習(xí)的研究成果。

不論是對(duì)于團(tuán)隊(duì)還是企業(yè)，TRAINS都能將所有內(nèi)容記錄在一個(gè)中央服務(wù)器中，并實(shí)現(xiàn)可視化和出處，這樣生產(chǎn)力就不會(huì)受到影響。TRAINS可以記錄和管理各種深度學(xué)習(xí)研究的模型負(fù)載，并且?guī)缀醪恍枰冻黾沙杀尽?/p>

我們專門設(shè)計(jì)了TRAINS，能夠輕松集成模型參數(shù)，團(tuán)隊(duì)可以保留現(xiàn)有的方法和實(shí)踐?？梢悦刻焓褂肨RAINS來(lái)增強(qiáng)團(tuán)隊(duì)協(xié)作，提升可視化程度，還可以用來(lái)將實(shí)驗(yàn)日志、輸出和數(shù)據(jù)收集到一個(gè)集中式的中央服務(wù)器上。

Demo地址：https://demoapp.trainsai.io （需要注冊(cè)）

究竟神奇在哪呢？以下是作者團(tuán)隊(duì)總結(jié)的TRAINS的主要特點(diǎn)。

TRAINS是我們解決機(jī)器學(xué)習(xí)/深度學(xué)習(xí)領(lǐng)域中與無(wú)數(shù)其他研究人員和開(kāi)發(fā)人員分享的問(wèn)題的方法：培訓(xùn)生產(chǎn)級(jí)深度學(xué)習(xí)模型是一個(gè)光榮而又混亂的過(guò)程。TRAINS通過(guò)關(guān)聯(lián)代碼版本控制、研究項(xiàng)目、性能指標(biāo)和模型出處來(lái)跟蹤和控制流程。

無(wú)縫兼容常用框架，一站式記錄所有模型數(shù)據(jù)

現(xiàn)在就能用

TRAINS免費(fèi)開(kāi)源，只需要兩行代碼即可完全集成。

可與其他常用工具一起使用

TRAINS與現(xiàn)有主要框架無(wú)縫集成，包括：PyTorch、TensorFlow、Keras和其他即將推出的框架支持Jupyter Notebook和PyCharm遠(yuǎn)程調(diào)試

TRAINS服務(wù)器和使用TRAINS python包的GPU訓(xùn)練機(jī)器的交互過(guò)程

記錄一切：讓機(jī)器學(xué)習(xí)實(shí)驗(yàn)真正可重復(fù)

自動(dòng)關(guān)聯(lián)模型+代碼+參數(shù)+初始權(quán)重的模型日志記錄

自動(dòng)在集中存儲(chǔ)上創(chuàng)建模型副本（支持共享文件夾、S3、GS，Azure即將推出！）

共享與合作

多用戶過(guò)程跟蹤與合作

中心服務(wù)器能夠整合日志、記錄和通用統(tǒng)計(jì)數(shù)據(jù)

提高生產(chǎn)力

全面的實(shí)驗(yàn)數(shù)據(jù)比對(duì)，代碼貢獻(xiàn)、初始權(quán)重、超參數(shù)和量度結(jié)果等。

組織與管理

管理并組織項(xiàng)目中的實(shí)驗(yàn)

查詢能力。通過(guò)結(jié)果量度對(duì)實(shí)驗(yàn)進(jìn)行分類和篩選。

更多特色

通過(guò)網(wǎng)頁(yè)應(yīng)用，利用遠(yuǎn)程機(jī)器終止實(shí)驗(yàn)

提供經(jīng)過(guò)現(xiàn)場(chǎng)測(cè)試的功能豐富的SDK，可滿足用戶的即時(shí)定制化需求

有關(guān)TRAINS機(jī)器學(xué)習(xí)模型管理平臺(tái)的更多詳情，可參閱GitHub資源：https://github.com/allegroai/trains