How to resolve runtime error due to size mismatch in PyTorch?

Question

I am trying to implement a simple autoencoder using PyTorch. My dataset consists of 256 x 256 x 3 images. I have built a torch.utils.data.dataloader.DataLoade

Answer 1

Your error:

size mismatch, m1: [76800 x 256], m2: [784 x 128]

says that previous layer output shape is not equal to next layer input shape

[76800 x 256], m2: [784 x 128] # Incorrect!
[76800 x 256], m2: [256 x 128] # Correct!