transformers
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
40 lines
1.1 KiB
40 lines
1.1 KiB
2 years ago
|
from auto_transformers import AutoTransformers
|
||
2 years ago
|
import onnx
|
||
2 years ago
|
|
||
2 years ago
|
f = open('onnx.csv', 'a+')
|
||
|
f.write('model_name, run op, save_onnx, check_onnx\n')
|
||
2 years ago
|
|
||
2 years ago
|
models = AutoTransformers.supported_model_names()[:1]
|
||
2 years ago
|
|
||
|
for name in models:
|
||
2 years ago
|
line = f'{name}, '
|
||
2 years ago
|
try:
|
||
|
op = AutoTransformers(model_name=name)
|
||
|
out1 = op('hello, world.')
|
||
2 years ago
|
line += 'success, '
|
||
2 years ago
|
except Exception as e:
|
||
2 years ago
|
line += 'fail\n'
|
||
|
f.write(line)
|
||
2 years ago
|
print(f'Fail to load op for {name}: {e}.')
|
||
2 years ago
|
continue
|
||
2 years ago
|
try:
|
||
|
op.save_model(format='onnx')
|
||
|
line += 'success, '
|
||
|
except Exception as e:
|
||
2 years ago
|
line += 'fail\n'
|
||
|
f.write(line)
|
||
2 years ago
|
print(f'Fail to save onnx for {name}: {e}.')
|
||
|
continue
|
||
|
try:
|
||
|
saved_name = name.replace('/', '-')
|
||
|
onnx_model = onnx.load(f'saved/onnx/{saved_name}.onnx')
|
||
|
onnx.checker.check_model(onnx_model)
|
||
|
line += 'success'
|
||
|
except Exception as e:
|
||
2 years ago
|
line += 'fail\n'
|
||
|
f.write(line)
|
||
2 years ago
|
print(f'Fail to check onnx for {name}: {e}.')
|
||
|
continue
|
||
|
line += '\n'
|
||
|
f.write(line)
|