diff --git a/configs/graph/default_graph.yaml b/configs/graph/default_graph.yaml index 327346cb6..ee08bc912 100644 --- a/configs/graph/default_graph.yaml +++ b/configs/graph/default_graph.yaml @@ -24,7 +24,7 @@ network: param_groups: fields: optimizer: - _target_: torch.optim.Adam + _target_: torch.optim.RAdam lr: 0.0005 scheduler: _target_: mattport.nerf.optimizers.ExponentialDecaySchedule diff --git a/mattport/nerf/renderers.py b/mattport/nerf/renderers.py index 86843ee5c..1ab8ec2e9 100644 --- a/mattport/nerf/renderers.py +++ b/mattport/nerf/renderers.py @@ -60,7 +60,7 @@ class RGBRenderer(nn.Module): TensorType[..., 3]: Outputs of rgb values. """ - rgb = self.combine_rgb(rgb, weights) + rgb = self.combine_rgb(rgb, weights, background_color=self.background_color) if not self.training: torch.clamp_(rgb, min=0.0, max=1.0) return rgb