issues
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
[ADD] Base class for `NLLLoss` & implement `MSELoss`
#252
Closed
jabader97
closed
2 years ago
f-dangel
commented
2 years ago
[ ] Only refactor
MSELoss
as
NLLLoss
in this PR
[ ] Revert to old implementation for
CrossEntropyLoss
[ ] Check for compatibility with devices (GPU)
[x] Think about strategies to reduce test run time
MSELossasNLLLossin this PRCrossEntropyLoss