Facebook和微軟發佈機器學習工具ONNX,PyTorch訓練的模型輕松轉到Caffe2


量子位 出品 | 公眾號 QbitAI



Facebook和微軟周四在各自的博客中發佈瞭ONNX,即“開放神經網絡轉換”工具。



簡單地說,這個新工具能把一種框架訓練的模型,轉換成另一種框架所需的格式。比如說,機器學習開發者可以將PyTorch訓練的模型轉換到Caffe2上,減少從研究到產品化所耗費的時間。



Facebook在博客中說,ONNX隻是第一步,他們的目標是建立一個開放的生態,讓AI開發者能在最先進的工具之間輕松流動,選擇最適合自己的框架。



在Facebook內部,研究和產品應用之間就有著很明顯的區隔,這傢公司一直有兩個機器學習團隊:FAIR(人工智能研究院)和AML(應用機器學習),FAIR專註於前沿性研究,而AML則關註如何將人工智能產品化。



深度學習框架的選擇是造成這種區別的關鍵。FAIR習慣於使用PyTorch。這種深度學習框架可以不考慮資源限制,在研究中取得漂亮的結果。



然而在現實世界中,大部分人都受限於智能手機和計算機的計算能力。當AML希望開發可以部署、可以規模化的產品時,通常會選擇Caffe2。Caffe2也是一種深度學習框架,但針對資源效率進行瞭優化。尤其是Caffe2Go特別關註瞭在性能不足的移動設備上優化機器學習模型。



Facebook和微軟的合作幫助研究者方便地將用PyTorch開發的模型轉換為Caffe2模型。通過降低兩種框架之間切換的障礙,Facebook和微軟可以推動研究的普及,加速整個商業化進程。



在這個月過完之前,微軟的CNTK、Facebook的Caffe2和PyTorch都將支持ONNX。





但距離“開放生態”的願景,ONNX還有很遠的路要走。並不是所有公司都在用PyTorch、Caffe2和CNTK,現在使用人數最多的框架,依然是Google的TensorFlow,而亞馬遜收編的MXNet、老牌的Theano、百度的paddle等等都各有一批用戶。

申請註冊商標台中

也正因為機器學習框架這麼多,功能大同小異,才讓ONNX這樣的工具有用武之地。



大部分機器學習框架的功能,都是讓開發者能更容易地建立、運行神經網絡的計算圖,而這些框架之間的核心區別,就在於它們用各不相同的格式來表示這些計算圖。



ONNX提供一種共享的模型表示,提升瞭AI框架之間的互操作性。





在科研范疇之外,其他人也在試圖將機器學習模型更方便地轉換為針對特定設備優化的形式。



例如,蘋果的CoreML就可以幫助開發台中商標註冊類別者轉換某些類型的模型。目前,CoreML還不支持TensorFlow,而且創建自定義轉換器的過程很復雜,轉換器最後有可能無法使用。



隨著谷歌和蘋果等公司加大對定制硬件上機器學習框架的優化,繼續關註互操作性非常重要。



目前,ONNX已經在Github上發佈,地址:https://github.com/onnx/onnx



— 完 —



活動報名





加入社群

台中商標申請查詢

量子位AI社群8群開始招募啦,歡迎對AI感興趣的同學,加小助手微信qbitbot3入群;



此外,量子位專業細分群(自動駕駛、CV、NLP、機器學習等)正在招募,面向正在從事相關領域的工程師及研究人員。



進群請加小助手微信號qbitbot3,並務必備註相應群的關鍵詞~通過審核後我們將邀請進群。(專業群審核較嚴,敬請諒解)

台灣電動床工廠 電動床

台灣電動床工廠 電動床

AUGI SPORTS|重機車靴|重機車靴推薦|重機專用車靴|重機防摔鞋|重機防摔鞋推薦|重機防摔鞋

AUGI SPORTS|augisports|racing boots|urban boots|motorcycle boots

arrow
arrow

    mbg945y8s4 發表在 痞客邦 留言(0) 人氣()