flash-attention/training/configs
2022-12-23 14:51:08 -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 Add smoothing for CrossEntropyParallel, rename to CrossEntropyLoss 2022-12-23 14:51:08 -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