nfnets_pytorch icon indicating copy to clipboard operation
nfnets_pytorch copied to clipboard

Published 20 hours ago • verified publisher icon benjs

AveragePool

Open bogdankjastrzebski opened this issue 3 years ago • 1 comments

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.

bogdankjastrzebski avatar May 16 '21 12:05 bogdankjastrzebski

Thanks for pointing out! I will fix it.

benjs avatar May 16 '21 15:05 benjs