@ptrblck: Thank you fo this. Pardon my ignorance, but, when you say:
I cannot say which approach is the right one, so you might want to run some experiments with and without the pooling.

do you mean I modify the mobilenet.py file and run experiments with and without this line:
x = nn.functional.adaptive_avg_pool2d(x, 1).reshape(x.shape[0], -1)
Also, what am I looking for in the results? Do you mean to say, I should measure accuracy in the two cases (with and withut the line?) Sorry about the naive questions, have not done this kind of experimenting before. I suspect what you are saying is that the approach is problem dependent (with and without pooling)?