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

dataloader

This commit is contained in:
shahules786 2022-10-19 16:39:31 +05:30
parent 38e0de689e
commit 23bee75ceb
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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

2
enhancer/data/dataset.py
View File

@ -263,7 +263,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], "->", start_time)
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
) )