OpenAI高調(diào)站隊:深度學(xué)習(xí)框架以后主要用PyTorch
乾明 發(fā)自 凹非寺
量子位 報道 | 公眾號 QbitAI
OpenAI發(fā)布最新消息,不再在各大AI框架中“搖擺”,而是支持PyTorch。
這家馬斯克參與創(chuàng)辦、微軟10億美元投資的頂級AI研究機構(gòu)表示,正在PyTorch上標(biāo)準(zhǔn)化自家深度學(xué)習(xí)框架。
“在過去,我們根據(jù)相對優(yōu)勢,在各種框架中實施特定的項目,”O(jiān)penAI在官方博客中表示,“現(xiàn)在我們選擇標(biāo)準(zhǔn)化,能使團(tuán)隊更容易創(chuàng)建和共享模型,并進(jìn)行優(yōu)化實現(xiàn)?!?/p>
與此同時,OpenAI也用實際行動表明了態(tài)度:發(fā)布了一個PyTorch 版本的開源教育資源,能夠讓人們更容易學(xué)習(xí)深度強化學(xué)習(xí)。
在博客文章中,他們也進(jìn)一步解釋了為什么選擇Pytorch:“主要原因是為了在GPU上提高我們的研究效率?!?/p>
并進(jìn)一步給Pytorch打Call:在PyTorch 中嘗試和執(zhí)行新的研究想法非常容易,能從幾周縮短到幾天。
按照OpenAI的說法,接下來他們將會主要使用PyTorch作為官方深度學(xué)習(xí)框架,在有特定技術(shù)原因的情況下,也會使用其他框架。
OpenAI表示,許多團(tuán)隊已經(jīng)完成了切換,并期待著在未來幾個月為 PyTorch 社區(qū)做出貢獻(xiàn)。
這一消息,也在社交媒體上引起了不少討論,Reddit的“機器學(xué)習(xí)”版面中,相關(guān)話題熱度已經(jīng)達(dá)到了530,110多人發(fā)表了看法。
排名最高的評論是:one of us,不少人認(rèn)為,OpneAI的行為是正常選擇的確,大量研究人員從TensorFlow轉(zhuǎn)投PyTorch已成趨勢。
比如CVPR 2019,只提到PyTorch的論文有280篇,但只提到TensorFlow的論文僅有125篇,而一年前這還是TensorFlow的天下。
不過也有不少人為TensorFlow撐場,稱研究中用Pytorch的確更好,但在生產(chǎn)中還是TensorFlow更有優(yōu)勢。
你怎么看?
OpenAI官方博客:
https://openai.com/blog/openai-pytorch/
討論現(xiàn)場:
https://www.reddit.com/r/MachineLearning/comments/ew8oxq/n_openai_switches_to_pytorch/