Given groups=1, weight of size [256, 65536, 7, 7], expected input[64, 3, 256, 256] to have 65536 channels, but got 3 channels instead

Question

class VAE(nn.Module):
    def __init__(self):
        super().__init__()

        self.encoder = nn.Sequential(
            nn.Conv2d(256 * 256, 256, kernel_size=7,