Thank you. I realized it afterwards.
Have one more question for you.(your opinion is good enough). What does AvergaePool does according to it and it is really useful? I am getting better results of my VGG CIFAR 10 classifier without that. Is that okay?
 self.avgpool = nn.AdaptiveAvgPool2d((8, 8))