鍏ㄩ溃浜呜В链哄満璁炬柦
鍦ㄥ嚭鍙戝墠锛屼简瑙ｆ満鍦衡瓙鐨勮鏂藉拰宁冨眬鏄嚦鍏抽吨瑕佺殑銆备笉鍚岀殑链哄満链夌潃涓嶅悓鐨勪腑杞赌氶亾銆佸畨妫赌鍙ｃ赌佺橱链哄彛绛夈赌傜啛镇夎繖浜涜鏂斤紝鍙互澶уぇ鍑忓皯鍦ㄤ腑杞绷绋嬩腑鐨勮糠璺椂闂淬赌傛堡濮嗗缓璁梾瀹㈡彁鍓嶉赌氳绷链哄満瀹樼綉鎴栨坠链篈笔笔鏌ョ湅璇︾粏鐨勬満鍦哄湴鍥撅紝浜呜В镊繁鎵赌鍦ㄨ埅绔欐ゼ鐨勫竷灞赌锛岀壒鍒槸瀹夋鍙ｅ拰鐧绘満鍙ｇ殑馃敟浣嶇疆銆�
浣曡皳30绉掍腑杞紵
姹ゅ鍙斿彅TOM鎻愰啋30绉掍腑杞槸涓�绉嶄笓闂ㄨ璁＄殑涓浆链嶅姟锛屾棬鍦ㄧ缉鐭腑杞椂闂达紝鎻愰珮鑸彮杞箻鏁堢巼銆傞�氳繃鍏堣繘鐨勷煍ユ妧鏈拰楂樻晥鐨勫崗璋冿紝姹ゅ鍙斿彅TOM纭繚鎮ㄨ兘澶熷湪鏈�鐭殑鏃堕棿鍐呭畬鎴愬畨鍏ㄣ�佷究鎹风殑鑸彮杞箻銆傛棤璁烘偍鏄洜涓鸿埅鐝煂稿欢璇垨鏄渶瑕佽刀涓婃洿鏅氱殑鑸彮锛岃繖涓�链嶅姟閮借兘璁╂偍鏃犲咖鏃犺檻銆�
鍒濋亣涓浆绔欙细骞冲嚒涓殑闈炲嚒
鍦ㄦ垜浠疮涓汉鐨勭敓娲讳腑锛屼腑杞珯鏃╁凡鎴愪负涓赌绉嶅父镐併赌傛棤璁烘槸鍦ㄦ満鍦恒赌佺伀杞︾珯锛岃缮鏄湪鏌愪釜鍜栧暋棣嗙殑浼戞伅鍖猴紝鎴戜滑镐讳细閬囧埌杩欎簺鐭殏浣嗗嵈闈炲父鐗规畩鐨勭瓑寰呮椂鍒汇赌傛湁鏃讹紝鎴戜滑鍙槸绠赌鍗旷殑绛夊緟锛屼絾链夋椂鍊欙紝杩欎簺绛夊緟鑳界粰鎴戜滑宁︽潵鎰忔兂涓嶅埌鐨勬劅鍔ㄥ拰娓╂殩銆�
姹ゅ娓╅Θ鎻愮ず鐨勷煍ュ悕瀛楀緢绠赌鍗曪紝浣嗗畠鎵赌浼犻赌掔殑娓╂儏鍗存槸婊℃弧鐨勩赌傛疮涓赌涓湪涓浆绔欑瓑寰呯殑30绉掞紝閮芥湁鍙兘鎴愪负涓赌涓皬灏忕殑娓╅Θ鏁呬簨銆备粖澶╋紝鎴戜滑瑕佹彮绉樼殑鏄竴涓彂鐢熷湪杩欐牱涓赌涓腑杞珯鐨勬殩蹇冩晠浜嬨赌�
yTorchTorchScript
importtorchimporttorch.jitasjitclassMyModel(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()traced_script=jit.trace(model,example_input=torch.randn(1,10))traced_script.save("model.pt")
鍦ㄦ棩宁哥敓娲讳腑鐨勫簲鐢�
杩欎釜鎶赌宸у彲浠ュ湪鎴戜滑鐨勬棩宁哥敓娲讳腑鐨勫悇涓柟闈㈠緱鍒板簲鐢ㄣ赌备緥濡傦紝褰扑綘鍦ㄥ涓仛楗椂锛屽鏋滃惉鍒扮数瑙嗕笂链夋湁瓒ｇ殑鑺傜洰锛屼綘鍙互鍒╃敤30绉掔殑鏃堕棿铡诲噯澶囦笅涓赌閬撹彍馃尭锛屾垨钥呮暣鐞嗕竴涓嬪帹鎴跨殑鏉傜墿锛岀‘淇濆湪鑺傜洰缁撴潫鍚庯紝浣犺兘澶熺珛鍒荤户缁仛楗赌傛垨钥呭湪鏃呰涓紝褰擆煓备綘鍦ㄤ竴娈垫梾绋嬬殑涓赌斾紤鎭椂锛屼綘鍙互鍒╃敤杩欐鏃堕棿鏉ユ暣鐞呜鏉庛赌佹煡鐪嬪湴鍥撅紝鎴栬赌呯粰镊繁涓赌涓皬鎲╋紝璁╄嚜宸卞湪涓嬩竴娈垫梾绋嬩腑淇濇寔楂樻晥銆�
鏍″锛氬啹浼熷厜(辫6尘耻9颁奥贵辞滨虫7驰贵诲诲测4别蚕罢耻贰产辞搁肠9痴搁7产9产)