(self, batch_size, num_threads, device_id, seed, image_dir)
| 76 | |
| 77 | class BasicPipeline(CommonPipeline): |
| 78 | def __init__(self, batch_size, num_threads, device_id, seed, image_dir): |
| 79 | super().__init__(batch_size, num_threads, device_id, seed, image_dir) |
| 80 | |
| 81 | def define_graph(self): |
| 82 | images, labels = self.load() |