Merge pull request #18 from ethanweber/tancik/losses

Add basic rendering losses
This commit is contained in:
Matthew Tancik authored and GitHub committed 2022-04-15 15:00:23 -07:00
commit eea1b61657
1 file changed
+9
+9
View File
@@ -0,0 +1,9 @@
"""
Collection of Losses
"""
from torch import nn
L1Loss = nn.L1Loss
MSELoss = nn.MSELoss
LOSSES = {"L1": L1Loss, "MSE": MSELoss}