Poisson losses bug fixes #31

Genentech / gReLU

gReLU is a python library to train, interpret, and apply deep learning models to DNA sequences.

https://genentech.github.io/gReLU/

MIT License

228 stars 23 forks source link

Poisson losses bug fixes #31 #32

Closed MuhammedHasan closed 3 months ago

avantikalal commented 3 months ago

This seems to be two separate changes - one to fix the PoissonMultinomial Loss and one to add device and dtype to all layers and metrics. Can we separate these into separate PRs?

avantikalal commented 3 months ago

I see that there is already a separate PR for the device/dtype update. Let's use this one only for the poisson loss fix.