Esempio n. 1
0
 def __init__(self):
     super().__init__(derivatives=CrossEntropyLossDerivatives())
Esempio n. 2
0
 def __init__(self):
     """Pass derivatives for ``torch.nn.CrossEntropyLoss`` module."""
     super().__init__(CrossEntropyLossDerivatives())