(args, dataset, split)
| 83 | |
| 84 | |
| 85 | def build_dataloader_func(args, dataset, split): |
| 86 | |
| 87 | if split == "train": |
| 88 | sampler = torch.utils.data.RandomSampler(dataset) |
| 89 | else: |
| 90 | sampler = torch.utils.data.SequentialSampler(dataset) |
| 91 | |
| 92 | dataloader = torch.utils.data.DataLoader( |
| 93 | dataset, |
| 94 | sampler=sampler, |
| 95 | batch_size=args.batchsize_per_gpu, |
| 96 | num_workers=args.dataset_num_workers, |
| 97 | worker_init_fn=my_worker_init_fn, |
| 98 | # add for meshgpt |
| 99 | drop_last = True, |
| 100 | ) |
| 101 | return sampler, dataloader |
| 102 | |
| 103 | |
| 104 | def build_dataset_func(args): |
no outgoing calls
no test coverage detected