Function create_dataloader

utils/data_utils.py:149–150

(dataset, args, shuffle=True)

def create_dataloader(dataset, args, shuffle=True):
    return DataLoader(dataset, batch_size=args.batch_size, shuffle=shuffle)

@dataclass
class DataCollatorForSupervisedDataset(object):