Merge pull request #106 from ethanweber/tancik/bug_fixes

RAdam + background color fix
This commit is contained in:
Matthew Tancik authored and GitHub committed 2022-05-24 11:30:33 -07:00
commit c3fa92b8c8
2 files changed
+2 -2

No files matched your search

+1 -1
View File
@@ -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
+1 -1
View File
@@ -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