search

aleximmer / Laplace

Laplace approximations for Deep Learning.
https://aleximmer.github.io/Laplace
MIT License
436 stars 63 forks source link

Implement support for BCEWithLogitsLoss #130

Open wiseodd opened 12 months ago

wiseodd commented 12 months ago

@AlexImmer @runame BackPACK has just been updated to support BCE loss [1]. We should add the support. Not sure if ASDL supports this, though. If it doesn't, how do we want to proceed?

[1] https://github.com/f-dangel/backpack/blob/master/changelog.md#160---2023-06-26

wiseodd commented 4 months ago

This can be done easily with torch.func and Curvlinops. Useful too, for img2img problems. Should be worked on once milestone 0.2 is done.

runame commented 3 months ago

BCEWithLogitsLoss support was just added to curvlinops in https://github.com/f-dangel/curvlinops/pull/99.