Commit 921cc06f authored by liuhy's avatar liuhy
Browse files

提交onnx模型

parent 4d0546cc
File added
......@@ -59,7 +59,7 @@ def validation(args):
print('Image: ' + args.imgpath + ' recongise result: '+ result)
if args.export_onnx:
print('export pytroch model to onnx model...')
print('export pytorch model to onnx model...')
onnx_input = torch.randn(1, 3, 24, 94, device=args.device)
torch.onnx.export(
model,
......
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