示例#1
0
 def make_dataloader(data, batch_size, shuffle, num_workers=0):
     dataloader = make_dataloader(data,
                                  batch_size,
                                  shuffle,
                                  num_workers,
                                  make_dataset=DatasetDurationSorted)
     return dataloader
示例#2
0
文件: cox.py 项目: MrHuff/Sumo-Net
 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