"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "b5b2b70f4a4bc8e608f16e6c9febddaa2e110d84"
Commit 7a8a2d7a authored by liuhy's avatar liuhy
Browse files

修正错误

parent a692a84b
...@@ -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