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')