鎬荤粨

閫氳繃浠ヤ笂姝ヰ煒庨�わ紝姹ゅ�嗗彅鍙斿笇鏈涙偍鑳藉�熷湪涓�杞�绔欏唴锛岄�氳繃楂樻晥鐨勬椂闂寸�＄悊鍜屽揩閫熷弽搴旓紝鎴愬姛瀹屾垚姣忎竴娆′腑杞�銆傛棤璁烘槸鍦伴搧銆佸叕浜よ繕鏄�鏈哄満锛岃繖濂楁柟娉曢兘鑳藉府鍔╂偍鍦ㄦ湁闄愮殑鏃堕棿鍐咃紝椤哄埄瀹屾垚杞�涔橈紝璁╂瘡涓�娆℃梾绋嬫洿鍔犺交鏉炬剦蹇�銆�

甯屾湜杩欎簺娓╅Θ鎻愮ず鑳藉�熷府鍔╂偍鍦ㄤ腑杞�绔欑殑姣忎竴娆＄粡鍘嗕腑锛屾洿鍔犺嚜濡傚拰杞绘澗銆傜�濇偍鏃呴�旀剦蹇�锛�

蹇冪悊璋冮��

闈㈠�圭敓娲讳腑鐨勫悇绉嶅帇鍔涘拰鎸戞垬锛屽績鐞嗚皟閫傛槸闈炲父閲嶈�佺殑銆傛堡濮嗗彅鍙斿缓璁�锛屾瘡澶╄姳涓�浜涙椂闂磋繘琛岃嚜鎴戝弽鎬濓紝鎬荤粨涓�澶╃殑寰楀け锛岃繖鏍疯兘甯�鍔╀綘鏇村ソ鍦拌�よ瘑鑷�宸憋紝璋冩暣蹇冩�併�傞�傚綋鐨勫ū涔愬拰

鍦ㄥ揩鑺傚�忕殑鐜颁唬鐢熸椿涓�锛屾垜浠�甯稿父浼氭劅鍒板帇鍔涘拰鐒﹁檻锛屼絾閫氳繃涓�浜涚畝鍗曠殑鏂规硶锛屾垜浠�鍙�浠ユ湁鏁堝湴绠＄悊鍜岀紦瑙ｈ繖浜涜礋闈㈡儏缁�銆傛堡濮嗗彅鍙斿皢涓哄ぇ瀹跺甫鏉ユ洿澶�30绉掔殑娓╅Θ鎻愮ず锛屽笇鏈涜兘涓轰綘鐨勬棩甯哥敓娲诲甫鏉ユ洿澶氭儸鎰忎笌渚筐煍ュ埄銆�

yTorchONNX

importtorchimportonnxclassMyModel(torch.nn.Module):def__init__(self):super(MyModel,self).__init__()self.fc=torch.nn.Linear(10,2)defforward(self,x):returnself.fc(x)model=MyModel()example_input=torch.randn(1,10)torch.onnx.export(model,example_input,"model.onnx")

鎬荤粨锛�

姹ゅ�嗗彅鍙旀俯棣ㄦ彁绀烘彁閱�30绉掍娇鐢ㄦ寚鍗楋紝涓烘偍鎻愪緵浜嗕竴绯诲垪绠�渚垮疄鐢ㄧ殑鐢熸椿灏忔妧宸с�傛棤璁烘槸瀹跺涵绠＄悊銆佸仴搴蜂繚鍏汇�佽储鍔＄�＄悊杩樻槸瀛︿範鎻愬崌锛岃繖浜涘皬璐村＋閮藉皢鎴愪负鎮ㄧ殑寰楀姏鍔╂墜銆傚彧闇�鍦ㄦ棩甯哥敓娲讳腑鎶曞叆30绉掔殑鏃堕棿锛屾偍灏嗕細鍙戠幇鐢熸椿鍙樺緱鏇村姞杞绘澗鎰夊揩銆�

甯屾湜杩欎簺灏忔妧宸ц兘澶熶负鎮ㄧ殑鏃ュ父鐢熸椿甯︷煋濇潵渚垮埄鍜屼箰瓒ｏ紝鎰挎偍姣忓ぉ閮借兘杩囦笂骞哥�忋�佸仴搴枫�佸厖瀹炵殑馃敟鐢熸椿锛�

瀛︿範鏂扮煡锛岃交鏉炬帉鎻�

鍦ㄥ揩閫熷彉鍖栫殑绀句細锛屾垜浠�闇�瑕佷笉鏂�瀛︿範鏂扮煡璇嗭紝鎻愬崌鑷�鎴戙�傛湁鏃跺�欏�︿範鏂扮煡璇嗛渶瑕佸ぇ閲忔椂闂村拰绮惧姏銆傛堡濮嗗彅鍙旀俯棣ㄦ彁绀烘彁閱�30s浣跨敤甯歌�侀棶棰樿В绛旓紝鑳藉�熷湪鐭�鐭�30绉掑唴锛屼负浣犳彁渚涘�︿範鏂扮煡璇嗙殑瑙ｅ喅鏂规�堬紝璁╀綘杞绘澗鎺屾彙鏂扮煡璇嗐��

渚嬪�傦紝褰撲綘鎯冲�︿範涓�闂ㄦ柊鐨勭紪绋嬭��瑷�鏃讹紝鎵嬫満灞忓箷涓婁細鑷�鍔ㄥ脊鍑烘俯棣ㄦ彁绀烘彁閱�30s浣跨敤甯歌�侀棶棰樿В绛旓紝缁欏嚭瀛︿範璺�绾垮拰寤鸿��锛氬�備綍蹇�閫熷叆闂�锛屽�備綍鏈夋晥澶嶄範銆傝繖鏍凤紝浣犲氨鑳藉湪鐭�鏃堕棿鍐呮帉鎻℃柊鐭ヨ瘑锛屾彁鍗囪嚜鎴戙��

鏍″�癸細鏌撮��(p6mu9CWFoIx7YFddy4eQTuEboRc9VR7b9b)