elituan

Results 1 issues of elituan

I want to extract the token embedding as shown in figure 11 of the paper. ![image](https://user-images.githubusercontent.com/34507382/192301545-e36560a5-4e54-44f2-8114-5dd1d785b2f5.png) However, when looking at the code, I see that the tokens are predicted by...