model_summary:
  _target_: pytorch_lightning.callbacks.RichModelSummary