Skip to content

Rebase, added option to plot MHAttention heads
Compare
Choose a tag to compare
@tatp22 tatp22 released this 29 Jun 05:49
· 54 commits to master since this release
0.11.0
547b7c9

Rebased the code so it looks better, and added the option to plot the
MHAttention module as well as the Linformer module

Assets 4
Loading