Export to ONNX Model possible?

[BoCiFOM]

Hello, I would like to export my trained model to onnx model.

Export Function call:

dummy_input2 = torch.LongTensor(2, 2) dummy_input2.requires_grad = False

torch.onnx.export(model.model,
(dummy_input0, dummy_input1, dummy_input2),
"Text-to-SQL.onnx",
input_names = ["input_ids", "input_mask", "segment_ids"],
output_names = ["output"]
)

“RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a parameter or input, or detaching the gradient”.

Do you some advices or a guide for exporting this model to onnx?

Best regards!