From 4a6b32d1dd449c4a3a4ca5f023a684254f08cba7 Mon Sep 17 00:00:00 2001 From: Si11ium Date: Fri, 15 May 2020 11:13:14 +0200 Subject: [PATCH] unlinken of already processed training files (seed dependent audio augmentation) --- datasets/binar_masks.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/datasets/binar_masks.py b/datasets/binar_masks.py index f587ba6..38477db 100644 --- a/datasets/binar_masks.py +++ b/datasets/binar_masks.py @@ -58,7 +58,11 @@ class BinaryMasksDataset(Dataset): # Delete File if one exists. if not self.use_preprocessed: for key in labeldict.keys(): - (self._mel_folder / (key.replace('.wav', '') + self.container_ext)).unlink(missing_ok=True) + try: + (self._mel_folder / (key.replace('.wav', '') + self.container_ext)).unlink() + except FileNotFoundError: + pass + return labeldict def __len__(self):