DPACMAN / configs /callbacks /model_summary.yaml
svincoff's picture
training works
29899b4
raw
history blame contribute delete
252 Bytes
# 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