(self, model, device_ids)
| 258 | trainer.test(task) |
| 259 | |
| 260 | def configure_ddp(self, model, device_ids): |
| 261 | model = DDP( |
| 262 | model, |
| 263 | device_ids=device_ids, |
| 264 | find_unused_parameters=True |
| 265 | ) |
| 266 | if dist.get_rank() != 0 and not hparams['debug']: |
| 267 | sys.stdout = open(os.devnull, "w") |
| 268 | sys.stderr = open(os.devnull, "w") |
| 269 | random.seed(hparams['seed']) |
| 270 | np.random.seed(hparams['seed']) |
| 271 | return model |
| 272 | |
| 273 | def training_end(self, *args, **kwargs): |
| 274 | return None |