(self)
| 135 | self.msg_client.send('[{}]: {}'.format(self.job_name, msg)) |
| 136 | |
| 137 | def create_dataset(self): |
| 138 | ginfo = self.ginfo |
| 139 | config = self.config |
| 140 | dataset_args = config.dataset['kwargs'] |
| 141 | dataset_args['ginfo'] = ginfo |
| 142 | self.dataset = datasets.dataset_entry(config.dataset) |
| 143 | dist.barrier() |
| 144 | |
| 145 | def create_dataloader(self): |
| 146 | config = self.config |