From 6028d918b6c00ba1e5ad0fd01da9d718826dfeb0 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 28 Oct 2022 10:17:51 +0530 Subject: [PATCH] debug ddp --- enhancer/cli/train.py | 1 + enhancer/cli/train_config/trainer/default.yaml | 2 +- enhancer/data/dataset.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py index 7d25af8..06ab966 100644 --- a/enhancer/cli/train.py +++ b/enhancer/cli/train.py @@ -45,6 +45,7 @@ def main(config: DictConfig): loss=parameters.get("loss"), metric=parameters.get("metric"), ) + print(dataset.train_dataloader().sampler) direction = model.valid_monitor checkpoint = ModelCheckpoint( diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 8bdf60f..0217553 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -23,7 +23,7 @@ limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 log_every_n_steps: 50 -max_epochs: 200 +max_epochs: 3 max_steps: -1 max_time: null min_epochs: 1 diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index b90f3b0..e2833b9 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -328,7 +328,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]) clean_segment = self.audio( file_dict["clean"], offset=start_time, duration=self.duration )