Forward pass not using the avgpool()

utkuozbulak / pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

MIT License

7.81k stars 1.49k forks source link

Forward pass not using the avgpool() #86

Closed viniciusarruda closed 4 years ago

viniciusarruda commented 4 years ago

Hi, at the forward_pass() functions, as in gradcam.py, the self.model.avgpool is not used. Why? At misc_functions.py, after loading the AlexNet model, you can print it and see that it is there, but didn't see any usage of it in the code.

utkuozbulak commented 4 years ago

Hey,

Yeah, should be there. Feel free to send a pr.

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.

utkuozbulak commented 4 years ago

Aha. You are right again. Thanks for the heads up.