fix stupid device variable

pull/3900/head
drryanhuang 9 months ago
parent 0108021b85
commit 83072c5f15

@ -787,14 +787,15 @@ class AudioSignal(
if device is None or "" == device: if device is None or "" == device:
return self return self
elif 'cpu' == device: elif 'cpu' == device:
device = paddle.to_tensor( self.audio_data = paddle.to_tensor(
self.audio_data, place=paddle.CPUPlace()) self.audio_data, place=paddle.CPUPlace())
elif 'gpu' == device or 'cuda' == device: elif 'gpu' == device or 'cuda' == device:
device = paddle.to_tensor( self.audio_data = paddle.to_tensor(
self.audio_data, place=paddle.CUDAPlace()) self.audio_data, place=paddle.CUDAPlace())
device = device.replace("cuda", else:
"gpu") if "cuda" in device else device device = device.replace("cuda",
self.audio_data = self.audio_data.to(device) "gpu") if "cuda" in device else device
self.audio_data = self.audio_data.to(device)
return self return self
def float(self): def float(self):

Loading…
Cancel
Save