Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

debug ddp

This commit is contained in:
shahules786 2022-10-28 10:17:51 +05:30
parent c321abe2ec
commit 6028d918b6
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
enhancer/cli/train.py
View File

@ -45,6 +45,7 @@ def main(config: DictConfig):
loss=parameters.get("loss"), loss=parameters.get("loss"),
metric=parameters.get("metric"), metric=parameters.get("metric"),
) )
print(dataset.train_dataloader().sampler)
direction = model.valid_monitor direction = model.valid_monitor
checkpoint = ModelCheckpoint( checkpoint = ModelCheckpoint(

2
enhancer/cli/train_config/trainer/default.yaml
View File

@ -23,7 +23,7 @@ limit_test_batches: 1.0
limit_train_batches: 1.0 limit_train_batches: 1.0
limit_val_batches: 1.0 limit_val_batches: 1.0
log_every_n_steps: 50 log_every_n_steps: 50
max_epochs: 200 max_epochs: 3
max_steps: -1 max_steps: -1
max_time: null max_time: null
min_epochs: 1 min_epochs: 1

2
enhancer/data/dataset.py
View File

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