convert pytorch model( pytorch_model.pt) to onnx

i have, PROJECT(folder) ├── pytorch_model.bin ├── bert_config.json └── vocab.txt

i would like to convert it to model.onnx

thanks a lot.

[TMVector]

@pumpkinband I’m guessing your code is erroring on the line dummy_output = model(dummy_input)? What I meant was: the dummy input needs to be valid input.

You need to provide a valid model input from somewhere:

import torch
import torch.onnx

model = torch.load('./pytorch_model.pt')

dummy_input = XXX # You need to provide the correct input here!

# Check it's valid by running the PyTorch model
dummy_output = model(dummy_input)
print("Input is valid")

# If the input is valid, then exporting should work
torch.onnx.export(model, dummy_input, "pytorch_model.onnx")

[pumpkinband]

i have graph.pbtxt i’ll try that !