search

f-dangel / backpack

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
https://backpack.pt/
MIT License
555 stars 55 forks source link

Issue 121 #250

Open jabader97 opened 2 years ago

jabader97 commented 2 years ago

Added test case for convolutional layer (when equal to linear layer) and commenting in KFAC methods to compare with KFAC paper.