fix stupid device variable

9 months ago · 83072c5f15
parent 0108021b85
commit 83072c5f15
1 changed files with 6 additions and 5 deletions
--- a/audio/audiotools/core/audio_signal.py
+++ b/audio/audiotools/core/audio_signal.py
@ -787,11 +787,12 @@ class AudioSignal(
            if device is None or "" == device:
                return self
            elif 'cpu' == device:
-                device = paddle.to_tensor(
+                self.audio_data = paddle.to_tensor(
                    self.audio_data, place=paddle.CPUPlace())
            elif 'gpu' == device or 'cuda' == device:
-                device = paddle.to_tensor(
+                self.audio_data = paddle.to_tensor(
                    self.audio_data, place=paddle.CUDAPlace())
+            else:
                device = device.replace("cuda",
                                        "gpu") if "cuda" in device else device
                self.audio_data = self.audio_data.to(device)