Fix run.py for new op.model

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
1 year ago · 88086ae2a8
1 changed files with 1 additions and 1 deletions
--- a/benchmark/run.py
+++ b/benchmark/run.py
@ -134,7 +134,7 @@ elif args.format == 'onnx':
        for o_n in output_names:
            dynamic_axes[o_n] = {0: 'batch_size', 1: 'sequence_length'}
        torch.onnx.export(
-            op.model,
+            op.model.model,
            tuple(inputs.values()),
            onnx_path,
            input_names=input_names,