We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bac3bbe commit 1fa974aCopy full SHA for 1fa974a
README.md
@@ -19,9 +19,10 @@ for param in model.parameters():
19
20
21
## Hessians
22
+(assuming ReLU activations, oherwise produces Gauss-Newton matrix)
23
24
```
25
autograd_hacks.backprop_hess(model(data), hess_type='CrossEntropy')
26
autograd_hacks.compute_hess(model)
27
print(param.hess) # print Hessian of param
-```
28
+```
0 commit comments