Fix server issues

remfx/datasets.py

@@ -190,9 +190,6 @@ class VocalSet(Dataset):
         self.chunk_size_in_sec = chunk_size_in_sec
         self.sample_rate = sample_rate
         self.mode = mode
-        import pdb
-
-        pdb.set_trace()
 
         mode_path = self.root / self.mode
         self.files = sorted(list(mode_path.glob("./**/*.wav")))

remfx/models.py

@@ -26,10 +26,10 @@ class FADLoss(torch.nn.Module):
         embds_background = []
         embds_eval = []
         for sample in audio_background:
-            embd = self.fad.model.forward(sample.T.detach().numpy(), self.sr)
+            embd = self.fad.model.forward(sample.T.cpu().detach().numpy(), self.sr)
             embds_background.append(embd.cpu().detach().numpy())
         for sample in audio_eval:
-            embd = self.fad.model.forward(sample.T.detach().numpy(), self.sr)
+            embd = self.fad.model.forward(sample.T.cpu().detach().numpy(), self.sr)
             embds_eval.append(embd.cpu().detach().numpy())
         embds_background = np.concatenate(embds_background, axis=0)
         embds_eval = np.concatenate(embds_eval, axis=0)

setup.py

@@ -46,6 +46,7 @@ setup(
         "auraloss",
         "pyloudnorm",
         "pedalboard",
+        "frechet_audio_distance",
     ],
     include_package_data=True,
     license="Apache License 2.0",