From 7dc53d6ddef2e56349188345e974eef517d3fa93 Mon Sep 17 00:00:00 2001 From: YuriHead Date: Thu, 13 Jul 2023 04:29:40 +0800 Subject: [PATCH] Debug --- modules/F0Predictor/RMVPEF0Predictor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/F0Predictor/RMVPEF0Predictor.py b/modules/F0Predictor/RMVPEF0Predictor.py index 755f61c..63412ae 100644 --- a/modules/F0Predictor/RMVPEF0Predictor.py +++ b/modules/F0Predictor/RMVPEF0Predictor.py @@ -71,9 +71,9 @@ class RMVPEF0Predictor(F0Predictor): vuv_vector = F.interpolate(vuv_vector[None,None,:],size=pad_to)[0][0] if f0.shape[0] <= 0: - return torch.zeros(pad_to, dtype=torch.float, device=x.device),vuv_vector.cpu().numpy() + return torch.zeros(pad_to, dtype=torch.float, device=x.device).cpu().numpy(),vuv_vector.cpu().numpy() if f0.shape[0] == 1: - return torch.ones(pad_to, dtype=torch.float, device=x.device) * f0[0],vuv_vector.cpu().numpy() + return (torch.ones(pad_to, dtype=torch.float, device=x.device) * f0[0]).cpu().numpy() ,vuv_vector.cpu().numpy() # 大概可以用 torch 重写? f0 = np.interp(time_frame, time_org, f0, left=f0[0], right=f0[-1])