While I am digging into this error further,  I found out that i am specifying in alexnext the following:
hiddens = [8,8]
flatten = 1800
while actually I am passing 224 * 224 = 11552 which is why it is throwing this error,
I resized the image to 32 * 32 and update the alextnet settings into:
hiddens = [32,32]
flatten = 1152
and solved this error
However this is affecting my alexnet network if I passed the input as 32 * 32 , and then I should be customising the network which I dont want to at this stage,
I am using this transformation and same image input 224.
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
same as this blog post AlexNet | PyTorch 1
anyway around this to pass the image size without resizing the image ?