eca55dc
1
2
3
4
5
6
# https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.callbacks.RichModelSummary.html model_summary: _target_: lightning.pytorch.callbacks.RichModelSummary max_depth: 1 # the maximum depth of layer nesting that the summary will include