search

OpenGVLab / SAM-Med2D

Official implementation of SAM-Med2D
Apache License 2.0
836 stars 80 forks source link

How to train on multi gpus? #34

Open KeyaoZhao opened 9 months ago

KeyaoZhao commented 9 months ago

Nice work! And I wonder how to train on multi gpus instead of single gpu? Can you provide the code plz? Thanks a lot!

Cjl-MedSeg commented 9 months ago

We have included multi-GPU training code in the source code of SAM-Med3D. If you need to run it on multiple cards, you can refer to the following link: https://github.com/uni-medical/SAM-Med3D/blob/main/train.py

biyuefeng commented 8 months ago

Hello, can Med2D train with multiple cards now? I failed to refer to the 3D multi card training.