Skip to content

GitLab

"docs/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "76b7d86a9a5c0c2186efa09c4a67b5f5666ac9e3"
Unverified Commit 7f9ecf67 authored by Ethan Weber's avatar Ethan Weber Committed by GitHub
Browse files

small fixes to occupancy code (#8)

parent ed327e36
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
nerfacc/occupancy_field.py
View file @ 7f9ecf67
...@@ -219,9 +219,9 @@ class OccupancyField(nn.Module): ...@@ -219,9 +219,9 @@ class OccupancyField(nn.Module):
"field during inference." "field during inference."
) )
if step % n == 0 and self.training: if step % n == 0 and self.training:
self.update( self._update(
step=step, step=step,
occ_threshold=occ_thre, occ_thre=occ_thre,
ema_decay=ema_decay, ema_decay=ema_decay,
warmup_steps=warmup_steps, warmup_steps=warmup_steps,
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment