Skip to content

Commit 1fa974a

Browse files
authored
Update README.md
1 parent bac3bbe commit 1fa974a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,10 @@ for param in model.parameters():
1919

2020

2121
## Hessians
22+
(assuming ReLU activations, oherwise produces Gauss-Newton matrix)
2223

2324
```
2425
autograd_hacks.backprop_hess(model(data), hess_type='CrossEntropy')
2526
autograd_hacks.compute_hess(model)
2627
print(param.hess) # print Hessian of param
27-
```
28+
```

0 commit comments

Comments
 (0)