to cpu
This commit is contained in:
parent
6eb905c1bb
commit
02192e5567
|
|
@ -131,8 +131,8 @@ class Pesq:
|
||||||
pesq_values.append(
|
pesq_values.append(
|
||||||
pesq(
|
pesq(
|
||||||
self.sr,
|
self.sr,
|
||||||
target_.squeeze().detach().numpy(),
|
target_.squeeze().detach().cpu().numpy(),
|
||||||
pred.squeeze().detach().numpy(),
|
pred.squeeze().detach().cpu().numpy(),
|
||||||
self.mode,
|
self.mode,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue