updated train for computing per token loss - #40

updated train for computing per token loss#40

adamimos wants to merge 1 commit into

mainadamimos/epsilon-transformers:mainfrom

per_token_lossadamimos/epsilon-transformers:per_token_lossCopy head branch name to clipboard