(cfg)
| 68 | |
| 69 | |
| 70 | def build_train_loader(cfg): |
| 71 | transforms = build_transforms(cfg, is_train=True) |
| 72 | dataset = build_dataset(cfg.INPUT.DATASET, cfg.INPUT.DATA_ROOT, transforms, "train") |
| 73 | return torch.utils.data.DataLoader( |
| 74 | dataset, |
| 75 | batch_size=cfg.INPUT.BATCH_SIZE_TRAIN, |
| 76 | shuffle=True, |
| 77 | num_workers=cfg.INPUT.NUM_WORKERS_TRAIN, |
| 78 | pin_memory=True, |
| 79 | drop_last=True, |
| 80 | collate_fn=collate_fn, |
| 81 | ) |
| 82 | |
| 83 | |
| 84 | def build_test_loader(cfg): |
no test coverage detected