From 02192e556729f38ecfff00c1415bb9d251b3b96b Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 22 Oct 2022 12:00:30 +0530 Subject: [PATCH] to cpu --- enhancer/loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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, ) )