fix argument error

autonomousvision · Jan 6, 2021 · 743c1f7 · 743c1f7
1 parent eaa2d6e
commit 743c1f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/submodules/nerf_pytorch/run_nerf.py b/submodules/nerf_pytorch/run_nerf.py
@@ -365,7 +365,7 @@ def config_parser():
     parser.add_argument("--netwidth", type=int, default=256, help='channels per layer')
     parser.add_argument("--netdepth_fine", type=int, default=8, help='layers in fine network')
     parser.add_argument("--netwidth_fine", type=int, default=256, help='channels per layer in fine network')
-    parser.add_argument("--N_samples", type=int, default=32*32*4, help='batch size (number of random rays per gradient step)')
+    parser.add_argument("--N_rand", type=int, default=32*32*4, help='batch size (number of random rays per gradient step)')
     parser.add_argument("--lrate", type=float, default=5e-4, help='learning rate')
     parser.add_argument("--lrate_decay", type=int, default=250, help='exponential learning rate decay (in 1000 steps)')
     parser.add_argument("--chunk", type=int, default=1024*32, help='number of rays processed in parallel, decrease if running out of memory')
@@ -545,7 +545,7 @@ def train():
             return
 
     # Prepare raybatch tensor if batching random rays
-    N_rand = args.N_samples
+    N_rand = args.N_rand
     use_batching = not args.no_batching
     if use_batching:
         # For random ray batching