导出动态轴的时候发现,明明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的名字,