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
549 stars 55 forks source link

[REF] Use `unfoldNd` package for input unfolding of convolutions #285

Closed f-dangel closed 1 year ago

f-dangel commented 1 year ago

Long ago, I extracted the im2col code from BackPACK's convolution utilities into a separate package, where it is documented, tested, and benchmarked in more detail. This PR removes code from the BackPACK repository and relies on the extracted package instead.