Open hongminglin08 opened 8 months ago
I used 2 Gpus for training, but halved the number of channels instead of the batch size. how should I modify it?
The current implementation does not include multi-GPU support. You can modify the code according to the PyTorch official DDP tutorial.
I used 2 Gpus for training, but halved the number of channels instead of the batch size. how should I modify it?