Browse Source
update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
2 changed files with
6 additions and
6 deletions
-
test_onnx.py
-
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='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) |
|
|
|
|
|
|
|
@ -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) |
|
|
|
|
