no cache again

popcornell · popcornell · commit 33b550b3d718 · 2020-12-30T22:18:53.000+01:00
diff --git a/speechbrain/lobes/augment.py b/speechbrain/lobes/augment.py
@@ -304,8 +304,6 @@ class EnvCorrupt(torch.nn.Module):
         A prepared csv file for loading room impulse responses.
     noise_csv : str
         A prepared csv file for loading noise data.
-    noise_cache : bool
-        Whether to cache noises.
     noise_num_workers : int
         Number of workers to use for loading noises.
     babble_speaker_count : int
@@ -340,7 +338,6 @@ def __init__(
         openrir_max_noise_len=None,
         reverb_csv=None,
         noise_csv=None,
-        noise_cache=False,
         noise_num_workers=0,
         babble_speaker_count=0,
         babble_snr_low=0,
@@ -386,7 +383,6 @@ def __init__(
             self.add_noise = AddNoise(
                 mix_prob=noise_prob,
                 csv_file=noise_csv,
-                cache=noise_cache,
                 num_workers=noise_num_workers,
                 snr_low=noise_snr_low,
                 snr_high=noise_snr_high,
diff --git a/speechbrain/processing/speech_augmentation.py b/speechbrain/processing/speech_augmentation.py
@@ -80,7 +80,6 @@ def __init__(
         csv_file=None,
         csv_keys=None,
         sorting="random",
-        cache=False,
         num_workers=0,
         snr_low=0,
         snr_high=0,
@@ -95,7 +94,6 @@ def __init__(
         self.csv_file = csv_file
         self.csv_keys = csv_keys
         self.sorting = sorting
-        self.cache = cache
         self.num_workers = num_workers
         self.snr_low = snr_low
         self.snr_high = snr_high
@@ -187,7 +185,6 @@ def _load_noise(self, lengths, max_length):
                     batch_size=batch_size,
                     num_workers=self.num_workers,
                     shuffle=(self.sorting == "random"),
-                    cache=self.cache,
                 )
                 self.noise_data = iter(self.data_loader)
 
