(dataset_name)
| 138 | |
| 139 | |
| 140 | def get_data(dataset_name): |
| 141 | isTrain = dataset_name == 'train' |
| 142 | augmentors = fbresnet_augmentor(isTrain) |
| 143 | return get_imagenet_dataflow( |
| 144 | args.data, dataset_name, BATCH_SIZE, augmentors) |
| 145 | |
| 146 | |
| 147 | def get_config(): |
no test coverage detected
searching dependent graphs…