model-summary.yaml 72 Bytes
Newer Older
zhangshao's avatar
zhangshao committed
1
2
model_summary:
  _target_: pytorch_lightning.callbacks.RichModelSummary