Fix issue with register_backward_hook

utkuozbulak / pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

MIT License

7.81k stars 1.49k forks source link

Fix issue with register_backward_hook #68

Closed stefanopini closed 4 years ago

stefanopini commented 4 years ago

Fix issue with register_backward_hook in guided_backprop.py (see issue #67)

utkuozbulak commented 4 years ago

Thanks for the PR. However, I'm still using torch version 0.4.1 myself as I have quite a lot of legacy code that depend on some stuff in this repository for my research and I do not have time to upgrade everything at the moment.

Since the whole repository is meant to work with this version (0.4.0, 0.4.1), making these changes will break the code for me. One day I might just upgrade all the code here to work with the latest version (at that time) but not yet.

holmosaint commented 4 years ago

Solved my problem, big thanks!