squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
40a25c8ee7
flash-attention/training/src/metrics
History
Tri Dao 71befc19e1 [Loss] Use flash_attn.losses.cross_entropy.CrossEntropyLoss
2022-12-31 22:43:28 -08:00
..
accuracy.py Release training code 2022-11-28 17:34:40 -08:00
num_tokens.py Release training code 2022-11-28 17:34:40 -08:00
perplexity.py [Loss] Use flash_attn.losses.cross_entropy.CrossEntropyLoss 2022-12-31 22:43:28 -08:00