i am seeing an error in my pytorch code can anyone help me to solve this error??

[tedzz12]

        padding=1
    ),
    nn.ReLU(),
    nn.MaxPool2d(kernel_size=2,stride=2)

)
self.classifier = nn.Sequential(
    nn.Flatten(),
    nn.Linear(
        in_features=hidden*16*16,
        out_features=output
    )
)

i put 16*16 in in_features but i still seeing this error : - ---------------------------------------------------------------------------
RuntimeError Traceback (most recent call last)
in <cell line: 11>()
10
11 for epoch in tqdm(range(epochs)):
---> 12 train_step(model=model_x.to(device),
13 dataloader=train_dataloader,
14 optimizer=optimizer,

6 frames
/usr/local/lib/python3.10/dist-packages/torch/nn/modules/linear.py in forward(self, input)
112
113 def forward(self, input: Tensor) -> Tensor:
--> 114 return F.linear(input, self.weight, self.bias)
115
116 def extra_repr(self) -> str:

RuntimeError: mat1 and mat2 shapes cannot be multiplied (32x4096 and 16384x2)
model_x = tinyvgg1(input=3,
hidden=64,
output=2
)