configs/callbacks/model_summary.yaml

# https://pytorch-lightning.readthedocs.io/en/latest/api/pytorch_lightning.callbacks.RichModelSummary.html

# Generates a summary of all layers in a LightningModule with rich text formatting.
# Look at the above link for more detailed information.
model_summary:
  _target_: lightning.pytorch.callbacks.RichModelSummary
  max_depth: 2 # The maximum depth of layer nesting that the summary will include. `-1` for all modules `0` for none.