mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-15 12:23:46 +08:00
10 lines
132 B
Python
10 lines
132 B
Python
"""
|
|
Collection of Losses
|
|
"""
|
|
from torch import nn
|
|
|
|
L1Loss = nn.L1Loss
|
|
MSELoss = nn.MSELoss
|
|
|
|
LOSSES = {"L1": L1Loss, "MSE": MSELoss}
|