Răsfoiți Sursa

[Quantization] run calib without grad

wanchen.swc 3 ani în urmă
părinte
comite
6ca0d1f54c
1 a modificat fișierele cu 2 adăugiri și 1 ștergeri
  1. 2 1
      funasr/export/export_model.py

+ 2 - 1
funasr/export/export_model.py

@@ -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)