(self, shuffle)
| 29 | |
| 30 | class BaseDataset(torch.utils.data.Dataset): |
| 31 | def __init__(self, shuffle): |
| 32 | super().__init__() |
| 33 | self.hparams = hparams |
| 34 | self.shuffle = shuffle |
| 35 | self.sort_by_len = hparams['sort_by_len'] |
| 36 | self.sizes = None |
| 37 | |
| 38 | @property |
| 39 | def _sizes(self): |