導出動態軸的時候發現,明明input.1就是onnx的名字,但是依然有警告說不是;
?
/miniconda3/envs/py38/lib/python3.8/site-packages/torch/onnx/utils.py:2078: UserWarning: Provided key input.1 for dynamic axes is not a valid input/output name warnings.warn( [ MainThread ] Input name: input.1
解決方法:明確指定輸入的名稱;
orch.onnx.export(model,dummy_input,"model_dynamic.onnx",input_names=['input.1'], # 明確指定輸入名稱dynamic_axes=dynamic_axes,opset_version=13 # 建議指定 opset 版本
)
猜測導出的時候,還不知道是什么名字,所以有警告,所以系統給了一個input.1的名字,