add input_size & batch_size to onnx_export

Boulaouaney · rwightman · commit a9c29a05de65 · 2023-09-21T14:46:25.000-07:00
diff --git a/onnx_export.py b/onnx_export.py
@@ -90,6 +90,8 @@ def main():
         check_forward=args.check_forward,
         training=args.training,
         verbose=args.verbose,
+        input_size=(3, args.img_size, args.img_size),
+        batch_size=args.batch_size,
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -90,6 +90,8 @@ def main():`
`90`	`90`	`check_forward=args.check_forward,`
`91`	`91`	`training=args.training,`
`92`	`92`	`verbose=args.verbose,`
	`93`	`+ input_size=(3, args.img_size, args.img_size),`
	`94`	`+ batch_size=args.batch_size,`
`93`	`95`	`)`
`94`	`96`
`95`	`97`