What type of GPU are you using and how long does it take

ByChelsea / VAND-APRIL-GAN

[CVPR 2023 Workshop] VAND Challenge: 1st Place on Zero-shot AD and 4th Place on Few-shot AD

162 stars 20 forks source link

What type of GPU are you using and how long does it take #25

Closed ZS-YANG closed 5 months ago

ZS-YANG commented 5 months ago

Thanks for your great work! By the way, what type of GPU are you using and how long does it take?🙌

ByChelsea commented 5 months ago

When training on VisA, fine-tuning takes about 5 minutes per epoch. Since the calculation of the PRO metric is very slow, if testing is conducted after each epoch, it would take about 27 (5+22) minutes per epoch. My experiments were conducted on an RTX 3090 (its power may have been limited at that time, but I don't remember).

ZS-YANG commented 5 months ago

Get it, thx ! 😝