Unverified Commit 7f9ecf67 authored by Ethan Weber's avatar Ethan Weber Committed by GitHub
Browse files

small fixes to occupancy code (#8)

parent ed327e36
...@@ -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