tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-14 11:54:01 +08:00
Code Activity

Fix tb and wandb crash (#769)

Browse Source
This commit is contained in:
Matthew Tancik authored and GitHub committed 2022-10-12 19:28:58 -07:00
1 parent 6b7cb70f0d
commit 5f2739e634
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
nerfstudio/models/nerfacto.py
+2 -2
View File
@@ -227,8 +227,8 @@ class NerfactoModel(Model):
loss_dict["interlevel_loss"] = self.config.interlevel_loss_mult * interlevel_loss(
outputs["weights_list"], outputs["ray_samples_list"]
)
assert metrics_dict is not None and "distortion" in metrics_dict
loss_dict["distortion_loss"] = self.config.distortion_loss_mult * metrics_dict["distortion"]
assert metrics_dict is not None and "distortion" in metrics_dict
loss_dict["distortion_loss"] = self.config.distortion_loss_mult * metrics_dict["distortion"]
return loss_dict
def get_image_metrics_and_images(