torch.onnx.export(
model,
(image, text), # 输入:图像 + tokenized 文本
text_fp32_onnx_path, # 输出 ONNX 文件
input_names=["image", "text"], # 输入节点名称
output_names=["image_features", "text_features"], # 输出节点名称
export_params=True, # 导出模型参数
opset_version=13, # ONNX opset 版本
verbose=True, # 打印详细信息
dynamic_axes={
"image": {0: "batch"}, # 动态 batch size(可选)
"text": {0: "batch"}, # 动态 batch size(可选)
},
)这样的写法是正确的吗