14 lines
295 B
Python
14 lines
295 B
Python
from torch import nn
|
|
|
|
from ml_lib.additions.losses import FocalLoss, FocalLossRob
|
|
|
|
|
|
class LossMixin:
|
|
|
|
absolute_loss = nn.L1Loss()
|
|
nll_loss = nn.NLLLoss()
|
|
bce_loss = nn.BCELoss()
|
|
ce_loss = nn.CrossEntropyLoss()
|
|
focal_loss = FocalLoss(None)
|
|
focal_loss_rob = FocalLossRob()
|