squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
a01d1213d7
flash-attention/training/configs
History
Tri Dao 71befc19e1 [Loss] Use flash_attn.losses.cross_entropy.CrossEntropyLoss
2022-12-31 22:43:28 -08:00
..
callbacks Release training code 2022-11-28 17:34:40 -08:00
datamodule Release training code 2022-11-28 17:34:40 -08:00
experiment [Loss] Use flash_attn.losses.cross_entropy.CrossEntropyLoss 2022-12-31 22:43:28 -08:00
logger Release training code 2022-11-28 17:34:40 -08:00
metrics Release training code 2022-11-28 17:34:40 -08:00
mode Release training code 2022-11-28 17:34:40 -08:00
model Release training code 2022-11-28 17:34:40 -08:00
optimizer Release training code 2022-11-28 17:34:40 -08:00
scheduler Release training code 2022-11-28 17:34:40 -08:00
task Release training code 2022-11-28 17:34:40 -08:00
trainer Release training code 2022-11-28 17:34:40 -08:00
config.yaml Release training code 2022-11-28 17:34:40 -08:00