squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
40a25c8ee7
flash-attention/training/src
History
Tri Dao 009a3e71ec [Training] Fix lightning _PATH import
2023-03-29 01:43:39 -07:00
..
callbacks Release training code 2022-11-28 17:34:40 -08:00
datamodules Release training code 2022-11-28 17:34:40 -08:00
distributed Release training code 2022-11-28 17:34:40 -08:00
metrics [Loss] Use flash_attn.losses.cross_entropy.CrossEntropyLoss 2022-12-31 22:43:28 -08:00
models/modules Release training code 2022-11-28 17:34:40 -08:00
optim Release training code 2022-11-28 17:34:40 -08:00
tasks Release training code 2022-11-28 17:34:40 -08:00
utils [Training] Fix lightning _PATH import 2023-03-29 01:43:39 -07:00
eval.py Release training code 2022-11-28 17:34:40 -08:00
train.py Release training code 2022-11-28 17:34:40 -08:00