Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nerfacc
Commits
b33d0d10
Commit
b33d0d10
authored
Sep 08, 2022
by
Ruilong Li
Browse files
small fix on doc
parent
e813bcaa
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
2 deletions
+6
-2
nerfacc/occupancy_field.py
nerfacc/occupancy_field.py
+6
-2
No files found.
nerfacc/occupancy_field.py
View file @
b33d0d10
...
...
@@ -175,16 +175,20 @@ class OccupancyField(nn.Module):
step
:
int
,
occ_thre
:
float
=
1e-2
,
ema_decay
:
float
=
0.95
,
warmup_steps
:
int
=
256
,
n
:
int
=
16
,
):
"""Update the field every n steps during training."""
if
not
self
.
training
:
raise
RuntimeError
(
"You should only call this function during training. Please call update() "
"directly if you want to update the field during inference."
"You should only call this function only during training. "
"Please call update() directly if you want to update the "
"field during inference."
)
if
step
%
n
==
0
and
self
.
training
:
self
.
update
(
step
=
step
,
occ_threshold
=
occ_thre
,
ema_decay
=
ema_decay
,
warmup_steps
=
warmup_steps
,
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment