(dataset, batch_size, root='./data', train=True)
| 125 | |
| 126 | |
| 127 | def get_indexed_tensor_loader(dataset, batch_size, root='./data', train=True): |
| 128 | if dataset == 'imagenet' or dataset == 'imagenet-mini': |
| 129 | return imagenet_utils.get_indexed_tensor_loader(dataset, batch_size, root, train) |
| 130 | |
| 131 | target_set = get_dataset(dataset, root=root, train=train) |
| 132 | target_set = data.IndexedTensorDataset(x=target_set.x, y=target_set.y) |
| 133 | |
| 134 | if train: |
| 135 | loader = data.Loader(target_set, batch_size=batch_size, shuffle=True, drop_last=True) |
| 136 | else: |
| 137 | loader = data.Loader(target_set, batch_size=batch_size, shuffle=False, drop_last=False) |
| 138 | |
| 139 | return loader |
| 140 | |
| 141 | |
| 142 | def get_poisoned_loader( |
nothing calls this directly
no test coverage detected