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