We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 04f2291 commit 30459e5Copy full SHA for 30459e5
onnx_diagnostic/torch_onnx/sbs.py
@@ -338,6 +338,16 @@ def post_process(obs):
338
print("final results")
339
df = pandas.DataFrame(results)
340
print(df)
341
+
342
+ A command line can also be run:
343
344
+ .. code-block:: bash
345
346
+ python -m onnx_diagnostic sbs -i <tensors>.input.pt \\
347
+ --ep <exported_program>.pt2 \\
348
+ -m <model>.onnx \\
349
+ -o results.xlsx \\
350
+ -v 1 --atol=0.1 --rtol=1
351
"""
352
assert callable(run_cls), f"run_cls={run_cls} not a callable"
353
str_kws = dict(with_shape=True, with_device=True, with_min_max=True)
0 commit comments