99999久久久久久亚洲,欧美人与禽猛交狂配,高清日韩av在线影院,一个人在线高清免费观看,啦啦啦在线视频免费观看www

熱線電話(huà)：13121318867

登錄
首頁(yè)大數(shù)據(jù)時(shí)代onnx解決不支持的pytorch算子？
onnx解決不支持的pytorch算子？
2023-04-07
收藏

ONNX（Open Neural Network Exchange）是一種開(kāi)放的、跨平臺(tái)的深度學(xué)習(xí)模型交換格式，它的目的是使得深度學(xué)習(xí)模型在不同的框架之間進(jìn)行轉(zhuǎn)換和移植變得更加容易。PyTorch 是一個(gè)廣泛使用的深度學(xué)習(xí)框架之一，但是由于 PyTorch 和 ONNX 的設(shè)計(jì)差異，有些 PyTorch 算子在 ONNX 中不被支持。本文將討論如何解決 ONNX 不支持的 PyTorch 算子。

  1. 使用ONNX 擴(kuò)展庫(kù)

ONNX 支持使用擴(kuò)展庫(kù)來(lái)擴(kuò)展其功能，這些擴(kuò)展庫(kù)包括 ONNX Runtime 和 ONNX Graphsurgeon。ONNX Runtime 提供了可用于 CPU 和 GPU 的高性能模型推理引擎，而 ONNX Graphsurgeon 則提供了一種將 ONNX 模型進(jìn)行修改和優(yōu)化的方法。通過(guò)這些擴(kuò)展庫(kù)，可以將 PyTorch 模型中不支持的算子轉(zhuǎn)換為 ONNX 模型中支持的算子。ONNX Runtime 和 ONNX Graphsurgeon 都是開(kāi)源項(xiàng)目，使用起來(lái)比較靈活，但需要用戶(hù)對(duì)深度學(xué)習(xí)模型的底層實(shí)現(xiàn)有一定的了解。

  1. 使用第三方工具

除了 ONNX 擴(kuò)展庫(kù)之外，還有一些第三方工具可以幫助我們解決 PyTorch 模型中不支持的算子。例如，MMdnn 是一個(gè)跨框架的深度學(xué)習(xí)模型轉(zhuǎn)換工具，支持從 PyTorch 轉(zhuǎn)換到多個(gè)其他框架，并且可以自動(dòng)處理不支持的算子。另外，TensorRT 是 NVIDIA 的一個(gè)高性能深度學(xué)習(xí)推理庫(kù)，可以將 PyTorch 模型轉(zhuǎn)換為 TensorRT 引擎，并且支持自定義算子。

  1. 手動(dòng)實(shí)現(xiàn)算子

如果沒(méi)有現(xiàn)成的工具可以解決 PyTorch 模型中不支持的算子，那么我們可以手動(dòng)實(shí)現(xiàn)這些算子，然后將其添加到 ONNX 模型中。這種方法需要一定的編程能力和對(duì)深度學(xué)習(xí)算法的理解，但是可以確保我們得到的 ONNX 模型與原始的 PyTorch 模型具有相同的功能。此外，ONNX 官方提供了一份詳細(xì)的開(kāi)發(fā)者指南，可以幫助我們了解如何實(shí)現(xiàn)自定義算子并將其添加到 ONNX 模型中。

  1. 重新設(shè)計(jì)模型

最后，如果以上方法都無(wú)法解決問(wèn)題，那么我們可能需要重新設(shè)計(jì)模型，以便使用 ONNX 支持的算子。在實(shí)際應(yīng)用中，我們應(yīng)該盡量避免使用不支持的算子，以便將深度學(xué)習(xí)模型在不同的框架之間進(jìn)行轉(zhuǎn)換和移植。

總結(jié)

在本文中，我們介紹了幾種解決 ONNX 不支持的 PyTorch 算子的方法。這些方法各有優(yōu)缺點(diǎn)，我們可以根據(jù)具體情況選擇最合適的方法。無(wú)論哪種方法，都需要對(duì)深度學(xué)習(xí)模型的底層實(shí)現(xiàn)有一定的了解，從而確保我們得到的 ONNX 模型具有相同的功能，并且可以在不同的框架之間進(jìn)行轉(zhuǎn)換和移植。

數(shù)據(jù)分析咨詢(xún)請(qǐng)掃描二維碼

若不方便掃碼，搜微信號(hào)：CDAshujufenxi

數(shù)據(jù)分析師資訊
更多

OK
客服在線
立即咨詢(xún)
客服在線
立即咨詢(xún)
') } function initGt() { var handler = function (captchaObj) { captchaObj.appendTo('#captcha'); captchaObj.onReady(function () { $("#wait").hide(); }).onSuccess(function(){ $('.getcheckcode').removeClass('dis'); $('.getcheckcode').trigger('click'); }); window.captchaObj = captchaObj; }; $('#captcha').show(); $.ajax({ url: "/login/gtstart?t=" + (new Date()).getTime(), // 加隨機(jī)數(shù)防止緩存 type: "get", dataType: "json", success: function (data) { $('#text').hide(); $('#wait').show(); // 調(diào)用 initGeetest 進(jìn)行初始化 // 參數(shù)1：配置參數(shù) // 參數(shù)2：回調(diào)，回調(diào)的第一個(gè)參數(shù)驗(yàn)證碼對(duì)象，之后可以使用它調(diào)用相應(yīng)的接口 initGeetest({ // 以下 4 個(gè)配置參數(shù)為必須，不能缺少 gt: data.gt, challenge: data.challenge, offline: !data.success, // 表示用戶(hù)后臺(tái)檢測(cè)極驗(yàn)服務(wù)器是否宕機(jī) new_captcha: data.new_captcha, // 用于宕機(jī)時(shí)表示是新驗(yàn)證碼的宕機(jī) product: "float", // 產(chǎn)品形式，包括：float，popup width: "280px", https: true // 更多配置參數(shù)說(shuō)明請(qǐng)參見(jiàn)：http://docs.geetest.com/install/client/web-front/ }, handler); } }); } function codeCutdown() { if(_wait == 0){ //倒計(jì)時(shí)完成 $(".getcheckcode").removeClass('dis').html("重新獲取"); }else{ $(".getcheckcode").addClass('dis').html("重新獲取("+_wait+"s)"); _wait--; setTimeout(function () { codeCutdown(); },1000); } } function inputValidate(ele,telInput) { var oInput = ele; var inputVal = oInput.val(); var oType = ele.attr('data-type'); var oEtag = $('#etag').val(); var oErr = oInput.closest('.form_box').next('.err_txt'); var empTxt = '請(qǐng)輸入'+oInput.attr('placeholder')+'！'; var errTxt = '請(qǐng)輸入正確的'+oInput.attr('placeholder')+'！'; var pattern; if(inputVal==""){ if(!telInput){ errFun(oErr,empTxt); } return false; }else { switch (oType){ case 'login_mobile': pattern = /^1[3456789]\d{9}$/; if(inputVal.length==11) { $.ajax({ url: '/login/checkmobile', type: "post", dataType: "json", data: { mobile: inputVal, etag: oEtag, page_ur: window.location.href, page_referer: document.referrer }, success: function (data) { } }); } break; case 'login_yzm': pattern = /^\d{6}$/; break; } if(oType=='login_mobile'){ } if(!!validateFun(pattern,inputVal)){ errFun(oErr,'') if(telInput){ $('.getcheckcode').removeClass('dis'); } }else { if(!telInput) { errFun(oErr, errTxt); }else { $('.getcheckcode').addClass('dis'); } return false; } } return true; } function errFun(obj,msg) { obj.html(msg); if(msg==''){ $('.login_submit').removeClass('dis'); }else { $('.login_submit').addClass('dis'); } } function validateFun(pat,val) { return pat.test(val); }