clean code

configs/lf_data/lf_africa.txt

@@ -9,7 +9,7 @@ no_reload = False
 testskip = 1
 
 ### TRAINING
-N_iters = 1250001
+N_iters = 500001
 N_rand = 512
 lrate = 0.0005
 lrate_decay_factor = 0.1

configs/tanks_and_temples/tat_training_truck.txt

@@ -9,7 +9,7 @@ no_reload = False
 testskip = 1
 
 ### TRAINING
-N_iters = 1250001
+N_iters = 500001
 N_rand = 1024
 lrate = 0.0005
 lrate_decay_factor = 0.1

nerf_network.py

@@ -109,7 +109,7 @@ class MLPNet(nn.Module):
 
         dim = 256 + self.input_ch_viewdirs
         for i in range(1):
-            rgb_layers.append(nn.Linear(dim, W))
+            rgb_layers.append(nn.Linear(dim, W // 2))
             rgb_layers.append(nn.ReLU())
             dim = W
         rgb_layers.append(nn.Linear(dim, 3))