speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

export model

Browse Source
This commit is contained in:
游雁 2023-02-14 16:44:31 +08:00
parent 1d4ab65c8b
commit 22ef23fb6a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/export/export_model.py
View File

@ -24,7 +24,7 @@ class ASRModelExportParaformer:
feats_dim=560, feats_dim=560,
onnx=False, onnx=False,
) )
logging.info("output dir: {}".format(self.cache_dir)) print("output dir: {}".format(self.cache_dir))
self.onnx = onnx self.onnx = onnx
@ -50,7 +50,7 @@ class ASRModelExportParaformer:
else: else:
self._export_torchscripts(model, verbose, export_dir) self._export_torchscripts(model, verbose, export_dir)
logging.info("output dir: {}".format(export_dir)) print("output dir: {}".format(export_dir))
def _export_torchscripts(self, model, verbose, path, enc_size=None): def _export_torchscripts(self, model, verbose, path, enc_size=None):