From 1ee3cab28112621e391f50f15cc9214d5ec36c0f Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Fri, 15 Apr 2022 14:58:23 -0700 Subject: [PATCH] Add basic rendering losses --- mattport/nerf/loss.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/mattport/nerf/loss.py b/mattport/nerf/loss.py index e69de29bb..d59f38dcd 100644 --- a/mattport/nerf/loss.py +++ b/mattport/nerf/loss.py @@ -0,0 +1,9 @@ +""" +Collection of Losses +""" +from torch import nn + +L1Loss = nn.L1Loss +MSELoss = nn.MSELoss + +LOSSES = {"L1": L1Loss, "MSE": MSELoss}