diff --git a/codes/data/audio/nv_tacotron_dataset.py b/codes/data/audio/nv_tacotron_dataset.py index 87bbb6d8..af909329 100644 --- a/codes/data/audio/nv_tacotron_dataset.py +++ b/codes/data/audio/nv_tacotron_dataset.py @@ -103,7 +103,7 @@ class TextMelLoader(torch.utils.data.Dataset): reload = False except: reload = True - if reload and self.max_mel_len != None and m.shape[-1] > self.max_mel_len: + if reload or (self.max_mel_len != None and m.shape[-1] > self.max_mel_len): print(f"Exception {index} {reload}") if not reload: print(f"mel_len:{m.shape[-1]} fname: {p}")