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

Double Scaling Bug Fix (#282)

Browse Source 
remove duplicate scaling.
This commit is contained in:
Alexander Kristoffersen 2022-08-16 13:25:41 -07:00 committed by GitHub
parent 6879a8a9bd
commit 261f1f3344
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nerfactory/models/base.py
View File

@ -175,10 +175,6 @@ class Model(nn.Module):
outputs=outputs, batch=batch, metrics_dict=metrics_dict, loss_coefficients=self.loss_coefficients
)
# scaling losses by coefficients.
for loss_name in loss_dict.keys():
if loss_name in self.loss_coefficients:
loss_dict[loss_name] *= self.loss_coefficients[loss_name]
return outputs, loss_dict, metrics_dict
def get_metrics_dict(self, outputs, batch) -> Dict[str, torch.Tensor]: