Modify pytorch IBA to hook model - Githubissues

BioroboticsLab / IBA

Information Bottlenecks for Attribution

MIT License

74 stars 9 forks source link

Modify pytorch IBA to hook model #4

Closed karl-schulz closed 4 years ago

karl-schulz commented 4 years ago

Use forward_hook for estimating and restricting instead of nn.Sequential.