fix rmvpe inference bugs

sandyzikun · Jul 21, 2023 · bb99d11 · bb99d11
1 parent 599add9
commit bb99d11
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]