Skip to content

Commit 5509766

Browse files
author
Frederik Rahbaek Warburg
committed
sign typo in crossentropy computation
1 parent 3998ff1 commit 5509766

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stochman/nnj.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1194,4 +1194,4 @@ def forward(self, x: Tensor) -> Tensor:
11941194

11951195
def _jacobian(self, x: Tensor, val: Tensor) -> Tensor:
11961196
jac = 0.5 / val
1197-
return jac
1197+
return jac

0 commit comments

Comments
 (0)