From 3d12eae17e068a7da3ab0aaafd1c711d1808952e Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 22 Jun 2022 14:28:31 +0800 Subject: [PATCH] update Signed-off-by: Jael Gu --- test_onnx.py | 6 +++--- test_torchscript.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/test_onnx.py b/test_onnx.py index 50d5ada..d8b018c 100644 --- a/test_onnx.py +++ b/test_onnx.py @@ -87,14 +87,14 @@ for name in models: out1 = op(data) f.write('success,') except Exception as e: - f.write('fail') + f.write('fail\n') print(f'Fail to load op for {name}: {e}') continue try: op.save_model(format='onnx') f.write('success,') except Exception as e: - f.write('fail') + f.write('fail\n') print(f'Fail to save onnx for {name}: {e}') continue try: @@ -103,7 +103,7 @@ for name in models: onnx.checker.check_model(onnx_model) f.write('success') except Exception as e: - f.write('fail') + f.write('fail\n') print(f'Fail to check onnx for {name}: {e}') continue f.write('\n') diff --git a/test_torchscript.py b/test_torchscript.py index b29d288..46df077 100644 --- a/test_torchscript.py +++ b/test_torchscript.py @@ -87,14 +87,14 @@ for name in models: out1 = op(data) f.write('success,') except Exception as e: - f.write('fail') + f.write('fail\n') print(f'Fail to load op for {name}: {e}') continue try: op.save_model(format='torchscript') f.write('success,') except Exception as e: - f.write('fail') + f.write('fail\n') print(f'Fail to save onnx for {name}: {e}') continue try: @@ -104,7 +104,7 @@ for name in models: assert (out1 == out2).all() f.write('success') except Exception as e: - f.write('fail') + f.write('fail\n') print(f'Fail to check onnx for {name}: {e}') continue f.write('\n')