Including avgpool in the AlexNet forward pass

utkuozbulak / pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

MIT License

7.81k stars 1.49k forks source link

Closed viniciusarruda closed 4 years ago

viniciusarruda commented 4 years ago

Resolves #86.

viniciusarruda commented 4 years ago

Sorry, I was using a more recent version of PyTorch. I found that previous versions do not have the avgpool.