diff --git a/enhancer/loss.py b/enhancer/loss.py index f2be8df..2150699 100644 --- a/enhancer/loss.py +++ b/enhancer/loss.py @@ -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, ) )