From e79995e664342788bbea8e3f6d0b1c538508aa2a Mon Sep 17 00:00:00 2001 From: Kai-46 Date: Thu, 15 Oct 2020 10:19:58 -0400 Subject: [PATCH] clean code --- configs/lf_data/lf_africa.txt | 2 +- configs/tanks_and_temples/tat_training_truck.txt | 2 +- nerf_network.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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))