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

[DOC] Sub-sampling use case #210

Closed f-dangel closed 3 years ago

f-dangel commented 3 years ago

Adds an example to the documentation that shows how to restrict BackPACK extensions to a subset of mini-batch samples.

Progress on #12.