dataloader

enhancer/cli/train_config/trainer/default.yaml

@@ -9,7 +9,7 @@ benchmark: False
 check_val_every_n_epoch: 1
 detect_anomaly: False
 deterministic: False
-devices: 2
+devices: 1
 enable_checkpointing: True
 enable_model_summary: True
 enable_progress_bar: True

enhancer/data/dataset.py

@@ -263,7 +263,7 @@ class EnhancerDataset(TaskDataset):
         return self.prepare_segment(*self._test[idx])
 
     def prepare_segment(self, file_dict: dict, start_time: float):
-
+        print(file_dict["clean"].split("/")[-1], "->", start_time)
         clean_segment = self.audio(
             file_dict["clean"], offset=start_time, duration=self.duration
         )