Code

feat: Add ModelSummary Callback#9344

Code

Merged

kaushikb11 merged 17 commits intomasterLightning-AI/pytorch-lightning:masterfrom

callback/model_summaryLightning-AI/pytorch-lightning:callback/model_summaryCopy head branch name to clipboard

Sep 10, 2021

Commits

Commits on Sep 6, 2021

• feat: Add ModelSummary Callback

  kaushikb11

  committed
  cdb8ebc

  Copy full SHA for cdb8ebc

Commits on Sep 9, 2021

• Address reviews

  kaushikb11

  committed
  e11d3a2

  Copy full SHA for e11d3a2

• Address reviews

  kaushikb11

  committed
  391724b

  Copy full SHA for 391724b

Commits on Sep 10, 2021

• Add ModelSummaryMode Enum

  kaushikb11

  committed
  11c93ba

  Copy full SHA for 11c93ba

• Update

  kaushikb11

  committed
  4937f9a

  Copy full SHA for 4937f9a

• Update ModelSummary callback

  kaushikb11

  committed
  dc135eb

  Copy full SHA for dc135eb

• Update for ModelSummaryMode

  kaushikb11

  committed
  e46ff0c

  Copy full SHA for e46ff0c

• Add test for ModelSummaryMode

  kaushikb11

  committed
  1019d5f

  Copy full SHA for 1019d5f

• Add tests for ModelSummary Callback

  kaushikb11

  committed
  5f2c5f4

  Copy full SHA for 5f2c5f4

• Update ModelSummary callback

  kaushikb11

  committed
  ee41317

  Copy full SHA for ee41317

• Merge branch 'master' into callback/model_summary

  kaushikb11

  authored
  d72aa44

  Copy full SHA for d72aa44

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 090bff1
  pre-commit-ci[bot]

  committed
  090bff1

  Copy full SHA for 090bff1

• Update ModelSummary callback

  kaushikb11

  committed
  2e748c4

  Copy full SHA for 2e748c4

• Merge branch 'callback/model_summary' of https://github.com/PyTorchLightning/pytorch-lightning into callback/model_summary

  kaushikb11

  committed
  9dc0ba2

  Copy full SHA for 9dc0ba2

• Fix tests

  kaushikb11

  committed
  e22adf4

  Copy full SHA for e22adf4

• Fix tests

  kaushikb11

  committed
  4baa620

  Copy full SHA for 4baa620

• Update pytorch_lightning/callbacks/model_summary.py

  Show description for a0760c8
  

  kaushikb11

  and

  ethanwharris

  authored
  a0760c8

  Copy full SHA for a0760c8