dummy_input = torch.randn(1, 3, 256, 256).to(device)
dummy_cam = torch.zeros(1, dtype=torch.long).to(device)
dummy_view = torch.zeros(1, dtype=torch.long).to(device)
torch.onnx.export(
model,
(dummy_input, dummy_cam, dummy_view),
“deit_transreid_veri.onnx”,
input_names=[‘input’, ‘cam_label’, ‘view_label’],
output_names=[“output”],
dynamic_axes={
‘input’: {0: ‘batch_size’},
‘cam_label’: {0: ‘batch_size’},
‘view_label’: {0: ‘batch_size’},
‘output’: {0: ‘batch_size’}
},
opset_version=13,
verbose=True,
)
print(“Model has been successfully exported to ONNX format!”)
when i run the code onnx export stop without error and no onnx file is generated
can you help me ?