diff --git a/configs/lf_data/lf_africa.txt b/configs/lf_data/lf_africa.txt
index 065d4e8..f9b1022 100644
--- a/configs/lf_data/lf_africa.txt
+++ b/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
diff --git a/configs/tanks_and_temples/tat_training_truck.txt b/configs/tanks_and_temples/tat_training_truck.txt
index 4d8d673..ee687a6 100644
--- a/configs/tanks_and_temples/tat_training_truck.txt
+++ b/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
diff --git a/nerf_network.py b/nerf_network.py
index c4aa52c..a40b65b 100644
--- a/nerf_network.py
+++ b/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))