Update src/infer_pack/predictor/RMVPE.py

src/infer_pack/predictor/RMVPE.py

@@ -1,7 +1,7 @@
+import sys, torch, numpy as np, traceback, pdb
 import torch.nn as nn
-import torch, numpy as np
+from time import time as ttime
 import torch.nn.functional as F
-from librosa.filters import mel
 
 
 class BiGRU(nn.Module):
@@ -245,6 +245,10 @@ class E2E(nn.Module):
                 nn.Dropout(0.25),
                 nn.Sigmoid(),
             )
+        else:
+            self.fc = nn.Sequential(
+                nn.Linear(3 * N_MELS, N_CLASS), nn.Dropout(0.25), nn.Sigmoid()
+            )
 
     def forward(self, mel):
         mel = mel.transpose(-1, -2).unsqueeze(1)
@@ -253,6 +257,9 @@ class E2E(nn.Module):
         return x
 
 
+from librosa.filters import mel
+
+
 class MelSpectrogram(torch.nn.Module):
     def __init__(
         self,