At some point `mnist_cnn.py` stopped training correctly. Convolution layer parameter initialization changed?