tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

RAdam + background color fix

Browse Source
This commit is contained in:
Matt Tancik 2022-05-24 11:28:05 -07:00
parent ade3b632de
commit 4cccc9110d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/graph/default_graph.yaml
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

2
mattport/nerf/renderers.py
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