From 23bee75ceb6e411efac562bf919cc6ad7dd49add Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 19 Oct 2022 16:39:31 +0530 Subject: [PATCH] dataloader --- enhancer/cli/train_config/trainer/default.yaml | 2 +- enhancer/data/dataset.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index ff24f6e..4fa4438 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/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 diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index 28f19a6..d140174 100644 --- a/enhancer/data/dataset.py +++ b/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 )