I just downloaded the 64x64 version, and I’m thinking about loading each batch into GPU memory (one at a time), to eliminate any potential bottleneck of a data transfer on each forward pass. What you think - is this worth doing? Each batch is 1.3GB so I’d be wasting ~10% of memory.