예제 #1
0
 def __init__(self):
     super().__init__(derivatives=CrossEntropyLossDerivatives())
예제 #2
0
파일: losses.py 프로젝트: f-dangel/backpack
 def __init__(self):
     """Pass derivatives for ``torch.nn.CrossEntropyLoss`` module."""
     super().__init__(CrossEntropyLossDerivatives())