Chatgpt couldn't give me a CNN for MNIST in pytorch that ran. Altough the code was OK, it always messed up the tensor sizes for each layer so it gave errors. I

Chatgpt couldn't give me a CNN for MNIST in pytorch that ran. Altough the code was OK, it always messed up the tensor sizes for each layer so it gave errors. It'd be interesting to test this with gpt 4.