20k iters

17e28de7 · Ruilong Li · b53f8a4e · 17e28de7 · 17e28de7
Commit 17e28de7 authored Mar 15, 2023 by Ruilong Li
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

examples/train_ngp_nerf.py examples/train_ngp_nerf.py +1 -1

examples/train_ngp_nerf_prop.py examples/train_ngp_nerf_prop.py +1 -1

No files found.
--- a/examples/train_ngp_nerf.py
+++ b/examples/train_ngp_nerf.py
@@ -60,7 +60,7 @@ if args.scene in MIPNERF360_UNBOUNDED_SCENES:
    from datasets.nerf_360_v2 import SubjectLoader

    # training parameters
-    max_steps = 100000
+    max_steps = 20000
    init_batch_size = 1024
    target_sample_batch_size = 1 << 18
    weight_decay = 0.0

--- a/examples/train_ngp_nerf_prop.py
+++ b/examples/train_ngp_nerf_prop.py
@@ -63,7 +63,7 @@ if args.scene in MIPNERF360_UNBOUNDED_SCENES:
    from datasets.nerf_360_v2 import SubjectLoader

    # training parameters
-    max_steps = 100000
+    max_steps = 20000
    init_batch_size = 4096
    weight_decay = 0.0
    # scene parameters