Skip to content

GitLab

"dist/git@developer.sourcefind.cn:cnjsdfcy/simbricks.git" did not exist on "f9a3a2e640f0b7c61bf9ee0a7b0e2559ca2f8405"
Commit 7a8a2d7a authored by liuhy's avatar liuhy
Browse files

修正错误

parent a692a84b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test.py
View file @ 7a8a2d7a
...@@ -91,7 +91,7 @@ def validation(args): ...@@ -91,7 +91,7 @@ def validation(args):
print('export pytorch model to onnx model...') print('export pytorch model to onnx model...')
onnx_input = torch.randn(1, 3, 24, 94, device=args.device) onnx_input = torch.randn(1, 3, 24, 94, device=args.device)
torch.onnx.export( torch.onnx.export(
model, lprnet,
onnx_input, onnx_input,
'LPRNet.onnx', 'LPRNet.onnx',
input_names=['input'], input_names=['input'],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment