Open jondo opened 5 years ago
rows = np.arange(i, min(i + batch_size, max_index)) #can be replaced by the simper rows = range(i, min(i + batch_size, max_index))
can you time it to see if there's any performance difference?
I'm using different (much higher dimensional) data, and I see no significant speed difference in a loop of several generator calls.