Lakonik / SSDNeRF

[ICCV 2023] Single-Stage Diffusion NeRF
https://lakonik.github.io/ssdnerf/
MIT License
438 stars 24 forks source link

different iteration numbers in unconditional generation and reconstruction #12

Closed byran-wang closed 1 year ago

byran-wang commented 1 year ago

@Lakonik Thank you for your great SSDNeRF work.

And I have one confusion with your work: Why the iteration number is 1M in unconditional generation training, which is significantly larger than 80K in reconstruction training.

Lakonik commented 1 year ago

Hi! Thank you for your interest in our work.

As mentioned in the paper, longer training schedule on a small dataset like ShapeNet Cars would cause the model to overfit training samples. Our temporary workaround is to use early stopping (at 80K) for a tradeoff between quality and generalization in reconstruction training.

byran-wang commented 1 year ago

Thanks for your kind answer.