Hardware / time needed for pretraining

nelson1425 / EfficientAD

Unofficial implementation of EfficientAD https://arxiv.org/abs/2303.14535

https://arxiv.org/abs/2303.14535

Apache License 2.0

273 stars 72 forks source link

Hardware / time needed for pretraining #6

Closed AlessandroPolidori closed 1 year ago

AlessandroPolidori commented 1 year ago

Hi, first of all thank you for your work and for making it completely public. I'm curious on the hardware you used for the pretraining on ImageNet, and on how much time did it take. Good job!

nelson1425 commented 1 year ago

Hello, I have pretrained on A100, A5000 or A6000. It took about 6 hours.