dataset stretch now optional

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