flash-attention/training/configs/callbacks/model-summary.yaml
2022-11-28 17:34:40 -08:00

3 lines
72 B
YAML

model_summary:
_target_: pytorch_lightning.callbacks.RichModelSummary