training with 2 gpus

[hongminglin08]

I used 2 Gpus for training, but halved the number of channels instead of the batch size. how should I modify it?

[dingfengshi]

The current implementation does not include multi-GPU support. You can modify the code according to the PyTorch official DDP tutorial.