Commit 6c09178
authored
Fix NaNs in
I was finding that `rasterizeTopContributingGaussianIdsForward` during
training could occasionally produce NaNs in its returned weights.
Logically I don't understand how a condition like this could be reached,
but initializing the output weights to zeros (instead of
`torch::empty`), mitigates this issue.
Signed-off-by: Jonathan Swartz <[email protected]>rasterizeTopContributingGaussianIdsForward weights (#400)1 parent 8899398 commit 6c09178
File tree
1 file changed
+1
-1
lines changed- src/fvdb/detail/ops/gsplat
1 file changed
+1
-1
lines changedLines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
387 | 387 | | |
388 | 388 | | |
389 | 389 | | |
390 | | - | |
| 390 | + | |
391 | 391 | | |
392 | 392 | | |
393 | 393 | | |
| |||
0 commit comments