search

mazurowski-lab / finetune-SAM

This is an official repo for fine-tuning SAM to customized medical images.
https://arxiv.org/abs/2404.09957
Apache License 2.0
114 stars 17 forks source link

ddp train failed #19

Open ZHIZIHUABU opened 3 weeks ago

ZHIZIHUABU commented 3 weeks ago

when i use ddp_splitgpu_train_finetune_noprompt.py , the error raised, “all tensors to be on the same device,but found at least two devices,cuda:1 and cuda:0, how can i solve it.

ZHIZIHUABU commented 3 weeks ago

Snipaste_2024-08-24_16-51-33

ZHIZIHUABU commented 3 weeks ago

Snipaste_2024-08-24_16-56-18 i change the code