That fixed that error, but now getting this one:
RuntimeError: size mismatch, m1: [32 x 6400], m2: [25600 x 256] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:290
I’m quite new to Pytorch and CNNs, so could you point me towards a resource that walks me through what the shape of each layer should be and what each component is?