tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

fix error with loss not defined

Browse Source
This commit is contained in:
machenmusik 2023-02-09 10:26:15 -05:00
parent 95793f5313
commit 47b1ef9898
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/model_components/losses.py
View File

@ -326,5 +326,5 @@ def robust_rgb_loss(rgb_gt,rgb_outputs,patch_size=16):
# final_w[:,
# (patch_size//2-patch_size//4):(patch_size//2+patch_size//4),
# (patch_size//2-patch_size//4):(patch_size//2+patch_size//4)] = weight_r8[:,None,None]
# loss = torch.mean(residuals.squeeze()*weight_r8[:,None,None])
loss = torch.mean(residuals.squeeze()*weight_r8[:,None,None])
return loss