Dataset rdy
This commit is contained in:
9
util/loss_mixin.py
Normal file
9
util/loss_mixin.py
Normal file
@@ -0,0 +1,9 @@
|
||||
from torch import nn
|
||||
|
||||
|
||||
class LossMixin:
|
||||
|
||||
absolute_loss = nn.L1Loss()
|
||||
nll_loss = nn.NLLLoss()
|
||||
bce_loss = nn.BCELoss()
|
||||
ce_loss = nn.CrossEntropyLoss()
|
||||
Reference in New Issue
Block a user