CUDA memory error during 2D classification

@Cwuz, are you currently running v3.2?