excellent work! May I ask how many and what type of GPUs did you use for training?

WisconsinAIVision / RISE

Domain Generalization through Distilling CLIP with Language Guidance

26 stars 2 forks source link

excellent work! May I ask how many and what type of GPUs did you use for training? #3

Closed StellarOdys2ey closed 1 year ago

OoDBag commented 1 year ago

Thanks. A single 2.4GB GPU is enough for train_rise.py, but train_rise_mix_teacher.py needs at least two. More GPUs will speed up training. Personally, I use 4 NVIDIA GeForce RTX 3090. Let me know if need more info about GPU and training.