[ADD] Support `BCEWithLogitsLoss` in `(Batch)DiagHessian`

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

Closed f-dangel closed 1 year ago

f-dangel commented 1 year ago

This PR adds support for BackPACK's DiagHessian and BatchDiagHessian extensions for the BCEWithLogitsLoss module.