diff --git a/test_onnx.py b/test_onnx.py index 4311655..bc92e41 100644 --- a/test_onnx.py +++ b/test_onnx.py @@ -15,7 +15,7 @@ for name in models: except Exception as e: line += 'fail\n' f.write(line) - print(f'Fail to load op for {name}: {e}.') + print(f'Fail to load op for {name}: {e}') continue try: op.save_model(format='onnx') @@ -23,7 +23,7 @@ for name in models: except Exception as e: line += 'fail\n' f.write(line) - print(f'Fail to save onnx for {name}: {e}.') + print(f'Fail to save onnx for {name}: {e}') continue try: saved_name = name.replace('/', '-') @@ -33,7 +33,7 @@ for name in models: except Exception as e: line += 'fail\n' f.write(line) - print(f'Fail to check onnx for {name}: {e}.') + print(f'Fail to check onnx for {name}: {e}') continue line += '\n' f.write(line) diff --git a/test_torchscript.py b/test_torchscript.py index 10eac79..12b5a0c 100644 --- a/test_torchscript.py +++ b/test_torchscript.py @@ -15,7 +15,7 @@ for name in models: except Exception as e: line += 'fail\n' f.write(line) - print(f'Fail to load op for {name}: {e}.') + print(f'Fail to load op for {name}: {e}') continue try: op.save_model(format='torchscript') @@ -23,7 +23,7 @@ for name in models: except Exception as e: line += 'fail\n' f.write(line) - print(f'Fail to save onnx for {name}: {e}.') + print(f'Fail to save onnx for {name}: {e}') continue try: saved_name = name.replace('/', '-') @@ -34,7 +34,7 @@ for name in models: except Exception as e: line += 'fail\n' f.write(line) - print(f'Fail to check onnx for {name}: {e}.') + print(f'Fail to check onnx for {name}: {e}') continue line += '\n' f.write(line)