Fixed a bug with the sequencing of the Linformer. Now should train properly.
0.12.1
A lm model is now available, for language modeling tasks
0.11.0
Rebased the code so it looks better, and added the option to plot the MHAttention module as well as the Linformer module
0.10.0
Check out pull request 7 to see the changes
0.9.0
Added an option to the linformer to compare it with full attention. Watch out, this takes O(n^2) time and space complexity now, where n is the sequence length
0.8.2
Added the option to save the visualization to a file