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
228 stars 23 forks source link

convert indices to int8 before stacking #21

Closed avantikalal closed 4 months ago

avantikalal commented 4 months ago

Currently, strings_to_indices converts all input strings to int64 arrays, stacks them, and then converts the whole array to int8. I changed this to create all the arrays as int8 in the first place so that the function uses less memory.