AveragePool

[bogdankjastrzebski]

Hi, noticed that the AveragePool ('pool' layer) is not used in forward function.
Instead, forward uses torch.mean. Removing the layer doesn't change pooling behavior.
I tried using this model as a feature extractor and was a bit confused for a moment.

[benjs]

Thanks for pointing out! I will fix it.