Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

to cpu

This commit is contained in:
shahules786 2022-10-22 12:00:30 +05:30
parent 6eb905c1bb
commit 02192e5567
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
enhancer/loss.py
View File

@ -131,8 +131,8 @@ class Pesq:
pesq_values.append(
pesq(
self.sr,
target_.squeeze().detach().numpy(),
pred.squeeze().detach().numpy(),
target_.squeeze().detach().cpu().numpy(),
pred.squeeze().detach().cpu().numpy(),
self.mode,
)
)