xuan-li / PAC-NeRF

Physics Augmented Continuum Neural Radiance Fields for Geometry-Agnostic System Identification
MIT License
268 stars 15 forks source link

ValueError when running the Cat example #6

Closed louhz closed 1 year ago

louhz commented 1 year ago

I am facing a float error when training the velocity for cat case

[W 07/06/23 15:30:56.874 2724704] [type_check.cpp:type_check_store@36] [$177661] Atomic add may lose precision: f32 <- f64 File "/DATA_EDS/louhz/PAC-NeRF/lib/engine/mpm_simulator.py", line 157, in svd_grad: self.F_tmp.grad[p] += self.backward_svd(self.U.grad[p].cast(ti.f64), ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ self.sig.grad[p].cast(ti.f64), self.V.grad[p].cast(ti.f64),

After look into the Taichi's discussion I find a issue:https://github.com/taichi-dev/taichi/issues/5059

Looks like Taichi did not fully support float64 if my understanding is correct

Do you have any suggestion about this ? Thanks!

xuan-li commented 1 year ago

This is only a warning.

louhz commented 1 year ago

This is only a warning. Thanks!