chapter 13 CNN MNIST model have a problem

the MINIST model code in your jupyter notebook viewer have a mistake.
the shape of pool3_flat layer here should be shape=[-1, pool3_fmaps * 14 * 14]

and how much storage you estimate this Mist model would need？Because I have GTX850M with 2GB video memory and it always throw ResourceExhaustedError exception.