flash-attention/training/configs/callbacks/model-summary.yaml

3 lines
72 B
YAML
Raw Normal View History

2022-11-29 09:31:19 +08:00
model_summary:
_target_: pytorch_lightning.callbacks.RichModelSummary