BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
555
stars
55
forks
source link
[ADD] `SqrtGGN{Exact, MC}` extension for `BatchNormNd` (eval mode) #244
Closed
f-dangel closed 2 years ago