tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 12:23:46 +08:00
Code Activity

Small device bug in mipnerf (#506)

Browse Source 
device bug in mipnerf.
This commit is contained in:
Alexander Kristoffersen authored and GitHub committed 2022-09-21 16:35:41 -07:00
1 parent 7b5a251a30
commit a1152be22b
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

nerfactory/models/mipnerf.py
+1 -1
View File
@@ -137,7 +137,7 @@ class MipNerfModel(Model):
return outputs
def get_loss_dict(self, outputs, batch, metrics_dict=None):
image = batch["image"]
image = batch["image"].to(self.device)
rgb_loss_coarse = self.rgb_loss(image, outputs["rgb_coarse"])
rgb_loss_fine = self.rgb_loss(image, outputs["rgb_fine"])
loss_dict = {"rgb_loss_coarse": rgb_loss_coarse, "rgb_loss_fine": rgb_loss_fine}
nerfactory/models/mipnerf_360.py
+1 -1
View File
@@ -147,7 +147,7 @@ class MipNerf360Model(Model):
return outputs
def get_loss_dict(self, outputs, batch, metrics_dict=None) -> Dict[str, torch.Tensor]:
image = batch["image"]
image = batch["image"].to(self.device)
rgb_loss_coarse = self.rgb_loss(image, outputs["rgb_coarse"])
rgb_loss_fine = self.rgb_loss(image, outputs["rgb_fine"])
loss_dict = {