fix: aligning depth None with internal FlopCounterMode.

GdoongMathew · GdoongMathew · commit 4b7eddfa2fd5 · 2025-06-01T16:18:21.000+08:00
diff --git a/src/lightning/pytorch/utilities/model_summary/model_summary.py b/src/lightning/pytorch/utilities/model_summary/model_summary.py
@@ -220,7 +220,7 @@ def __init__(self, model: "pl.LightningModule", max_depth: int = 1) -> None:
         self._flop_counter = FlopCounterMode(
             mods=None if _TORCH_GREATER_EQUAL_2_4 else self._model,
             display=False,
-            depth=max_depth + 1,
+            depth=max_depth + 1 if max_depth >= 0 else None,
         )
 
         self._max_depth = max_depth

Original file line number	Diff line number	Diff line change
`@@ -220,7 +220,7 @@ def __init__(self, model: "pl.LightningModule", max_depth: int = 1) -> None:`
`220`	`220`	`self._flop_counter = FlopCounterMode(`
`221`	`221`	`mods=None if _TORCH_GREATER_EQUAL_2_4 else self._model,`
`222`	`222`	`display=False,`
`223`		`- depth=max_depth + 1,`
	`223`	`+ depth=max_depth + 1 if max_depth >= 0 else None,`
`224`	`224`	`)`
`225`	`225`
`226`	`226`	`self._max_depth = max_depth`