OpenGait release(pre-beta version).
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
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
|
||||
Reference in New Issue
Block a user