Assignment MNIST Dataset facing problem in train_mnist function

You posted this in the General Discussions category. Please move it to the relevant course category as described here and our mentors will be happy to help you.