onnx export

游雁

2023-02-27 6da6a7847fe78d125068dbb0d019f8a7e50fcf53

onnx export

 funasr/export/export_model.py

@@ -11,8 +11,8 @@
import numpy as np
import random

torch_version = float(".".join(torch.__version__.split(".")[:2]))
assert torch_version > 1.9
# torch_version = float(".".join(torch.__version__.split(".")[:2]))
# assert torch_version > 1.9

class ASRModelExportParaformer:
    def __init__(self, cache_dir: Union[Path, str] = None, onnx: bool = True):