Man, thank you for all your help and quick responses. But it’s still not working. I think it’s worth mentioning I’m using a channel size of 8. Even if I change nn.Linear to be nn.Linear(256 X 8 X 8) it gives me: size mismatch, m1: [1 x 147456], m2: [16384 x 20]