From cb8b09e085bdfb5599ccd1a862912bc0b7e4f41c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 20 二月 2024 16:42:09 +0800
Subject: [PATCH] update

---
 funasr/datasets/audio_datasets/preprocessor.py |    7 ++++---
 1 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/funasr/datasets/audio_datasets/preprocessor.py b/funasr/datasets/audio_datasets/preprocessor.py
index c2e27bf..a3ba3a5 100644
--- a/funasr/datasets/audio_datasets/preprocessor.py
+++ b/funasr/datasets/audio_datasets/preprocessor.py
@@ -26,9 +26,10 @@
 			return waveform
 		speed = random.choice(self.speed_perturb)
 		if speed != 1.0:
-			waveform, _ = torchaudio.sox_effects.apply_effects_tensor(
-				torch.tensor(waveform).view(1, -1), fs, [['speed', str(speed)], ['rate', str(fs)]])
-			waveform = waveform.view(-1)
+			with torch.no_grad():
+				waveform, _ = torchaudio.sox_effects.apply_effects_tensor(
+					torch.tensor(waveform).view(1, -1), fs, [['speed', str(speed)], ['rate', str(fs)]])
+				waveform = waveform.view(-1)
 			
 		return waveform
 

--
Gitblit v1.9.1