MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.26k stars 538 forks source link

Can this model be trained on multi GPUs ? #70

Closed ChibisukeDragon closed 2 years ago

ChibisukeDragon commented 3 years ago

I see that nnUNET has DP and DDP versions, but nnUNET++ doesn't... How to train this model on multi gpus?