BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
549
stars
55
forks
source link
[ADD] Kronecker Hessian approximations for `ConvTranspose{1,2,3}d` #292
Closed
f-dangel closed 1 year ago
This PR generalizes the Kronecker approximations for convolution to transpose convolution.