SAM-Adapter-PyTorch icon indicating copy to clipboard operation
SAM-Adapter-PyTorch copied to clipboard
verified publisher icon tianrun-chen

How do I train with a single multi-card?

Open demoyu123 opened this issue 1 year ago • 2 comments

demoyu123 avatar May 10 '24 08:05 demoyu123

@tianrun-chen Please can you provide loadddptrain.py, I can't find the file in your project, thanks!

demoyu123 avatar May 10 '24 08:05 demoyu123

你可以试试以下代码：CUDA_VISIBLE_DEVICES=2,3 torchrun --standalone --nnodes=1 --nproc_per_node=2 train.py --config configs/demo.yaml

xupinggl avatar Nov 07 '24 11:11 xupinggl