Dataset rdy

This commit is contained in:
Steffen Illium
2021-02-16 10:18:04 +01:00
parent 151b22a2c3
commit 7edd3834a1
11 changed files with 350 additions and 15 deletions

9
util/loss_mixin.py Normal file
View 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()