當前位置: 華文世界 > 科技

微軟神經網路引擎ONNX獲推1.17更新,支持瀏覽器訓練模型

2024-02-15科技

IT之家 2 月 14 日訊息,據微軟官方新聞稿,微軟日前為旗下神經網路推論引擎「ONNX Runtime」推出了 1.17 版本,主要加入了網頁端訓練功能,允許開發者在瀏覽器中訓練模型。

據介紹,微軟在 ONNX Runtime 1.8 版本中,首度為引擎推出了瀏覽器內推論功能,該功能利用 WebGL 和 WebGPU 進行推論,允許開發者呼叫瀏覽器跨平台部署機器學習模型。

而在最新的 ONNX Runtime 1.17 中,微軟選擇進一步強化瀏覽器在訓練過程中起到的作用,允許開發者使用瀏覽器訓練模型。

微軟提到,相關基於瀏覽器的網頁訓練功能主要由 WebAssembly 和 JavaScript API 兩部份組成,其中 WebAssembly 用於轉換程式碼,JavaScript API 用於管理 WebAssembly 與緩沖區轉換過程 。

▲ 圖源 微軟官方新聞稿

IT之家註意到,由於瀏覽器不能直接允許 C 語言程式碼,因此微軟使用 WebAssembly 將程式碼轉換為能夠在瀏覽器中高效執行的.wasm 二進制檔,以便於開發者在瀏覽器中進行模型訓練。

微軟同時提到,目前基於瀏覽器的模型訓練方式僅支持 CPU 單執行緒運算 ,開發者同時需要使用 ONNX 工具生成必要的執行環境,之後才能進行訓練。