Hi,

I am referring to your code. I have some confusion related to forward function.

I'm trying to implement VGG11 using the CIFAR10 dataset but got some dimension errors.

Can you please explain how did you make the dimensions inside forward function like this?

torch.zeros(batch_size, 64, 32, 32, device=device))

When I tried this using VGG11 , got this error?

It means my dimensions are not correct.

Below is the snippet of my forward function: