junzhezhang / shape-inversion

[CVPR 2021] Unsupervised 3D Shape Completion through GAN Inversion
MIT License
131 stars 21 forks source link

How long does it take to train? #5

Closed gxwangupc closed 3 years ago

gxwangupc commented 3 years ago

I feel very interested in your work. I'd like to know what GPUs you were using and how long does it take to train.

junzhezhang commented 3 years ago

Hi, In the GAN Inversion stage, for each of the test example, it takes roughly half a minute on a single V100. The framework supports distributed.

For pertaining, it takes less than 10 hours on 4 V100.

gxwangupc commented 3 years ago

Thank you very much.