yvquanli/TrimNet

Visualizing weights

Closed this issue · 2 comments

gurugecl commented

Hi, would it be at all possible to please share the code to visualize the atomic attention weights for model interpretability?

yvquanli commented

May you can try to adapt some code from this file https://github.com/yvquanli/GLAM/blob/main/src_1gp/visualize_gp.py

gurugecl commented

Ok great, thank you very much for sharing that!