Update with dynamic axes

README.md

@@ -58,8 +58,12 @@ torch.onnx.export(colbert_query_encoder,
                 f="query_encoder_colbert.onnx",
                 input_names = input_names,
                 output_names = output_names,
+                dynamic_axes = {
+                    "input_ids": {0: "batch"},
+                    "attention_mask": {0: "batch"},
+                    "contextual": {0: "batch"},
+                },
                 opset_version=11)
-
 ```
 
 # Representing the model on Vespa.ai