From bb99d1124957aebeb855384094dcd30341079f2e Mon Sep 17 00:00:00 2001 From: "llc1995@sina.com" Date: Fri, 21 Jul 2023 19:48:50 +0800 Subject: [PATCH] fix rmvpe inference bugs --- encoder/rmvpe/inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/encoder/rmvpe/inference.py b/encoder/rmvpe/inference.py index 2227221..e48f29c 100644 --- a/encoder/rmvpe/inference.py +++ b/encoder/rmvpe/inference.py @@ -12,7 +12,7 @@ def __init__(self, model_path, hop_length=160): self.resample_kernel = {} model = E2E0(4, 1, (2, 2)) ckpt = torch.load(model_path) - model.load_state_dict(ckpt['model']) + model.load_state_dict(ckpt['model'], strict=False) model.eval() self.model = model self.mel_extractor = MelSpectrogram(N_MELS, SAMPLE_RATE, WINDOW_LENGTH, hop_length, None, MEL_FMIN, MEL_FMAX) @@ -21,7 +21,7 @@ def __init__(self, model_path, hop_length=160): def mel2hidden(self, mel): with torch.no_grad(): n_frames = mel.shape[-1] - mel = F.pad(mel, (0, 32 * ((n_frames - 1) // 32 + 1) - n_frames), mode='reflect') + mel = F.pad(mel, (0, 32 * ((n_frames - 1) // 32 + 1) - n_frames), mode='constant') hidden = self.model(mel) return hidden[:, :n_frames]