hello，how to multi-gpu training？

MontaEllis / Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.

MIT License

860 stars 196 forks source link

hello，how to multi-gpu training？ #19

Closed zhang5957 closed 3 years ago

MontaEllis commented 3 years ago

We would add DDP in the future. You can modify: https://github.com/MontaEllis/Pytorch-Medical-Segmentation/blob/22c1ac83646024428c7198ebf614436a5e97b058/main.py#L2 to simple multi gpu mode.

zhang5957 commented 3 years ago

We would add DDP in the future. You can modify:

https://github.com/MontaEllis/Pytorch-Medical-Segmentation/blob/22c1ac83646024428c7198ebf614436a5e97b058/main.py#L2

to simple multi gpu mode.

when i was training，my dice is always 0.it‘s my data problem?

MontaEllis commented 3 years ago

Yes

zhang5957 commented 3 years ago

Yes

thanks