How to make the model run in DDP

uni-medical / STU-Net

The largest pre-trained medical image segmentation model (1.4B parameters) based on the largest public dataset (>100k annotations), up until April 2023.

Apache License 2.0

276 stars 23 forks source link

How to make the model run in DDP #12

Open okhezi0001 opened 1 year ago

okhezi0001 commented 1 year ago

i want run this model in DDP,how can i get it

Ziyan-Huang commented 1 year ago

We haven't tried running the model in DDP as of now, so we are not entirely sure about the specifics.