def make_dataloader(data, batch_size, shuffle, num_workers=0): dataloader = make_dataloader(data, batch_size, shuffle, num_workers, make_dataset=DatasetDurationSorted) return dataloader
def make_dataloader(data, batch_size, shuffle, num_workers=0): dataloader = tt.make_dataloader( data, batch_size, shuffle, num_workers, make_dataset=models.data.DurationSortedDataset) return dataloader