(self, image_size, batch_size)
| 49 | |
| 50 | class SimpleDataManager(DataManager): |
| 51 | def __init__(self, image_size, batch_size): |
| 52 | super(SimpleDataManager, self).__init__() |
| 53 | self.batch_size = batch_size |
| 54 | self.trans_loader = TransformLoader(image_size) |
| 55 | |
| 56 | def get_data_loader(self, data_file, aug): # parameters that would change on train/val set |
| 57 | transform = self.trans_loader.get_composed_transform(aug) |
nothing calls this directly
no test coverage detected