mirror of
https://github.com/shivammehta25/Matcha-TTS.git
synced 2026-02-04 17:59:19 +08:00
6 lines
252 B
YAML
6 lines
252 B
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
|