update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
2 years ago · 013c6e7fc3
1 changed files with 1 additions and 1 deletions
--- a/clip_vision.py
+++ b/clip_vision.py
@ -68,7 +68,7 @@ class ClipVision(NNOperator):
            return False
        x = torch.randn((1, 3, 224, 224))

-        torch.onnx.export(self.model, x, output_file, input_names=['INPUT0'],
+        torch.onnx.export(self.model.model, x, output_file, input_names=['INPUT0'],
                          output_names=['OUTPUT0'],  dynamic_axes={'INPUT0': [0]})
        return True