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}