Просмотр исходного кода

[Quantization] run calib without grad

wanchen.swc 3 лет назад
Родитель
Сommit
6ca0d1f54c
1 измененных файлов с 2 добавлено и 1 удалено
  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)