squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
8f4cd4c16b
flash-attention/training/configs/callbacks/model-summary.yaml
Tri Dao 0bf5e50038 Release training code
2022-11-28 17:34:40 -08:00

3 lines
72 B
YAML
Raw Blame History

model_summary:
_target_: pytorch_lightning.callbacks.RichModelSummary
Reference in New Issue View Git Blame Copy Permalink