9 lines
164 B
Python
9 lines
164 B
Python
from torch import nn
|
|
|
|
|
|
class LossMixin:
|
|
|
|
absolute_loss = nn.L1Loss()
|
|
nll_loss = nn.NLLLoss()
|
|
bce_loss = nn.BCELoss()
|
|
ce_loss = nn.CrossEntropyLoss() |