Does the weight file support exporting onnx? about open-groundingdino HOT 2 CLOSED

You can consider using torch.onnx.export for conversion.

from open-groundingdino.

@longzw1997 @BIGBALLON

I am trying to export to onnx and having troubles with the **kw in the forward function.

input = (image[None], {"caption":[caption]})
torch.onnx.export(model, input, "gdinot-1.8m-odvg.onnx", input_names=["samples", "kw"], output_names=["pred_logits", "pred_boxes"])

It keep is not passing the dict into **kw. I have also tried passing the caption using the targets argument but it also generated another error.
Please advice.

from open-groundingdino.