dataset stretch now optional

2020-05-15 11:06:23 +02:00
parent 09f9dd9131
commit 9ae9143544
4 changed files with 21 additions and 12 deletions
--- a/util/module_mixins.py
+++ b/util/module_mixins.py
@ -142,6 +142,7 @@ class BinaryMaskDatasetMixin:
            **dict(
                # TRAIN DATASET
                train_dataset=BinaryMasksDataset(self.params.root, setting=V.DATA_OPTIONS.train,
+                                                 use_preprocessed=self.params.use_preprocessed,
                                                 mixup=self.params.mixup, stretch_dataset=self.params.stretch,
                                                 mel_transforms=mel_transforms_train, transforms=aug_transforms),
                # VALIDATION DATASET