issues
search
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
221
stars
23
forks
source link
Add support for half precision (bf16 and fp16) and on device model creation
#26
Closed
MuhammedHasan
closed
3 months ago
MuhammedHasan
commented
3 months ago
PR changes:
Changed positional embedding device dtype logic to support half precision
add dtype and device argument to class
loss function and metrics should always be calculated with float32
PR changes: