I understand. Would I benefit from using a binary cross entropy? I only have two classes. In this case would I be restricted to long too?
The idea of precompute is good, the only problem is that now my dataset is already a 150 Gb of space…