RuntimeError: size mismatch, m1: [a x b], m2: [c x d]
img = img.view(1, -1) # converts [3 x 256 x 256] to 1 x 196608 output = model(img)