Closed Yiir-1 closed 1 year ago
Hi! We train the segmentation model on 8 A100 GPUs, and it takes about 20 and 25 hours to train PolarFormer and PolarFormer-T, respectively. (A100 GPU is not a must for the training of PolarFormer, we simply use it to accelerate training. You can just reduce the batch size and train it on 3090 or V100 GPUs.)
Thanks for reply !
Hello!Thanks for your good work ! Could you please explain what GPU you used to train models, and how long does it take?