clean code

This commit is contained in:
Kai-46 2020-10-15 10:19:58 -04:00
parent 5994239676
commit e79995e664
3 changed files with 3 additions and 3 deletions

View file

@ -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

View file

@ -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

View file

@ -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))