CoOp icon indicating copy to clipboard operation
CoOp copied to clipboard

training speed

Open jianpursuit opened this issue 2 years ago • 1 comments

Thank you for your contribution. I found that the training is slower when using multi-gpus (e.g., 8 gpus) than single gpu. Do you know why is it and how to speed up the training process?

jianpursuit avatar Jan 22 '22 10:01 jianpursuit

using DistributedDataParallel can gain a significant increase in the speed but this would require big changes to the underlying Dassl package

my bad, I didn't consider DistributedDataParallel when designing Dassl

so for this CoOp code I'd highly suggest you use a single gpu

KaiyangZhou avatar Jan 22 '22 12:01 KaiyangZhou