(m)
| 585 | |
| 586 | if opt.distribution_type == 'multi': # Multiple processes, single GPU per process |
| 587 | def _transform_(m): |
| 588 | return nn.parallel.DistributedDataParallel( |
| 589 | m, device_ids=[gpu], output_device=gpu) |
| 590 | |
| 591 | torch.cuda.set_device(gpu) |
| 592 | model.cuda(gpu) |
nothing calls this directly
no outgoing calls
no test coverage detected