search

3DTopia / LGM

[ECCV 2024 Oral] LGM: Large Multi-View Gaussian Model for High-Resolution 3D Content Creation.
https://me.kiui.moe/lgm/
MIT License
1.71k stars 118 forks source link

How does LGM support `fp16` training for `diff-gaussian-rasterization`? #58

Closed thucz closed 4 months ago

thucz commented 5 months ago

Hi! I wonder which code you revised to support fp16 training in the original diff-gaussian-rasterization?

yifanlu0227 commented 4 months ago

Hi, have you figured it out?

thucz commented 4 months ago

In fact, their diff-gaussian-rasterization does not support fp16. they use float32 in this module and fp16 in their neural networks.