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

[Quantization] run calib without grad

Browse Source
This commit is contained in:
wanchen.swc 2023-03-16 18:53:57 +08:00
parent 4ffd5655da
commit 6ca0d1f54c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/export/export_model.py
View File

@ -76,7 +76,8 @@ class ASRModelExportParaformer:
feats, feats_len = self.load_feats(self.audio_in)
for i, (feat, len) in enumerate(zip(feats, feats_len)):
print("debug, iter: {}".format(i))
m(feat, len)
with torch.no_grad():
m(feat, len)
else:
dummy_input = model.get_dummy_inputs()
m(*dummy_input)