コード例 #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())