search

lutao2021 / BrightDreamer

BrightDreamer: Generic 3D Gaussian Generative Framework for Fast Text-to-3D Synthesis
https://vlislab22.github.io/BrightDreamer/
MIT License
83 stars 2 forks source link

How many epochs does the vehicle dataset need to be trained? #8

Closed Mike001-wq closed 1 month ago

Mike001-wq commented 1 month ago

Do all 600,000 training epochs in vehicle.sh need to be completed? I conduct experiment on 4 A40 and 800+hours are needed, much more than twice the 36 hours.

lutao2021 commented 1 month ago

Sorry for this misunderstanding.

Actually, we didn't train the model 600,000 iterations. After you find there is no quality improvement, you could stop it manually.

Mike001-wq commented 1 month ago

Sorry for this misunderstanding.

Actually, we didn't train the model 600,000 iterations. After you find there is no quality improvement, you could stop it manually.

Thank you very much for your reply. How many epochs can we stop after? Roughly speaking?

lutao2021 commented 1 month ago

Sorry for this misunderstanding. Actually, we didn't train the model 600,000 iterations. After you find there is no quality improvement, you could stop it manually.

Thank you very much for your reply. How many epochs can we stop after? Roughly speaking?

Maybe 200 epoches for IF, and another 100 epoches for IF+IF2.

It depends on your batch_size and learning rate.