From 77d5146ca4b64ecfe66b77e3822eb5302cfcaa12 Mon Sep 17 00:00:00 2001 From: Kai Zhang Date: Thu, 15 Oct 2020 23:30:57 -0400 Subject: [PATCH] minor fix --- nerf_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerf_network.py b/nerf_network.py index a40b65b..41eabd9 100644 --- a/nerf_network.py +++ b/nerf_network.py @@ -111,7 +111,7 @@ class MLPNet(nn.Module): for i in range(1): rgb_layers.append(nn.Linear(dim, W // 2)) rgb_layers.append(nn.ReLU()) - dim = W + dim = W // 2 rgb_layers.append(nn.Linear(dim, 3)) rgb_layers.append(nn.Sigmoid()) # rgb values are normalized to [0, 1] self.rgb_layers = nn.Sequential(*rgb_layers)