INNER CODE UNIT · Python

build_dataloader_func

OpenMeshLab/MeshXL · main.py:85

def build_dataloader_func(args, dataset, split):
    
    if split == "train":
        sampler = torch.utils.data.RandomSampler(dataset)
    else:
        sampler = torch.utils.data.SequentialSampler(dataset)
    
    dataloader = torch.utils.data.DataLoader(
        dataset,
        sampler=sampler,
        batch_size=args.batchsize_per_gpu,
        num_workers=args.dataset_num_workers,
        worker_init_fn=my_worker_init_fn,
        # add for meshgpt
        drop_last = True,
    )
    return sampler, dataloader

View source record →