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

1
funasr/export/export_model.py
View File

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