I get this error while running mnist_cnn_one_iteration.py. Kindly shed some clarity on how to remove this error