search

i-machine-think / diagNNose

diagNNose is a Python library that facilitates a broad set of tools for analysing hidden activations of neural models.
https://diagnnose.readthedocs.io
MIT License
81 stars 8 forks source link

AWD-LSTM fix #81

Closed Kalsir closed 3 years ago

Kalsir commented 3 years ago

Current code cannot properly handle multilayer AWD-LSTM models because it incorrectly assumes that the model only has a single layer. This is a small fix for that.