梁峻豪
14 lines
373 B
Python
14 lines
373 B
Python
import torch.nn as nn
|
|
from utils import Odict
|
|
|
|
class BasicLoss(nn.Module):
|
|
def __init__(self, loss_term_weights=1.0):
|
|
super(BasicLoss, self).__init__()
|
|
|
|
self.loss_term_weights = loss_term_weights
|
|
self.pair_based_loss = True
|
|
self.info = Odict()
|
|
|
|
def forward(self, logits, labels):
|
|
raise NotImplementedError
|