MUnique:

Is it possible that the size of the input array in each epoch (200MB) cause the OOM problem?


4000*200*200*20*16bits ~= 6GB, so no mysteries there. 200mb is zipped .npz size, you can’t train a network on that representation.