修正错误

7a8a2d7a · liuhy · a692a84b · 7a8a2d7a
Commit 7a8a2d7a authored Mar 06, 2023 by liuhy
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test.py test.py +1 -1

No files found.
--- a/test.py
+++ b/test.py
@@ -91,7 +91,7 @@ def validation(args):
        print('export pytorch model to onnx model...')
        onnx_input = torch.randn(1, 3, 24, 94, device=args.device)
        torch.onnx.export(
-            model,
+            lprnet,
            onnx_input,
            'LPRNet.onnx',
            input_names=['input'],