I was using Ctrl-F in Chrome for the words on the training data (Shakespeare texts). With Teansformer models it's quite easy to print out top Query-Key pairs t

I was using Ctrl-F in Chrome for the words on the training data (Shakespeare texts). With Teansformer models it's quite easy to print out top Query-Key pairs to debug what happened, but that was not my intention.