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

cer tool

Browse Source
This commit is contained in:
游雁 2023-03-23 10:01:32 +08:00
parent 8e4fa38ad3
commit 129cfcd9f2
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

@ -14,7 +14,7 @@ from funasr.utils.types import str2bool
# torch_version = float(".".join(torch.__version__.split(".")[:2])) # torch_version = float(".".join(torch.__version__.split(".")[:2]))
# assert torch_version > 1.9 # assert torch_version > 1.9
class ASRModelExportParaformer: class ModelExport:
def __init__( def __init__(
self, self,
cache_dir: Union[Path, str] = None, cache_dir: Union[Path, str] = None,
@ -240,7 +240,7 @@ if __name__ == '__main__':
parser.add_argument('--calib_num', type=int, default=200, help='calib max num') parser.add_argument('--calib_num', type=int, default=200, help='calib max num')
args = parser.parse_args() args = parser.parse_args()
export_model = ASRModelExportParaformer( export_model = ModelExport(
cache_dir=args.export_dir, cache_dir=args.export_dir,
onnx=args.type == 'onnx', onnx=args.type == 'onnx',
quant=args.quantize, quant=args.quantize,