Commit feb757b
Allard Hendriksen
Fix weights access when pruning in pytorch 1.4
Patch based on discussion at:
https://discuss.pytorch.org/t/forward-pre-hooks-not-called-after-applying-nn-uitls-prune-methods/69497
This patch has been tested by Richard Schoonhoven.1 parent c09f48d commit feb757b
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
176 | 176 | | |
177 | 177 | | |
178 | 178 | | |
179 | | - | |
| 179 | + | |
180 | 180 | | |
181 | 181 | | |
0 commit comments