configs/callbacks/model_summary.yaml

# https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.callbacks.RichModelSummary.html

model_summary:
  _target_: lightning.pytorch.callbacks.RichModelSummary
  max_depth: 3 # the maximum depth of layer nesting that the summary will include