Revert "fix: aligning depth None with internal FlopCounterMode."

GdoongMathew · GdoongMathew · commit 61b9abc8b85f · 2025-06-01T16:35:54.000+08:00
This reverts commit 4b7eddf.
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 if max_depth >= 0 else None,
+            depth=max_depth + 1,
         )
 
         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 if max_depth >= 0 else None,`
	`223`	`+ depth=max_depth + 1,`
`224`	`224`	`)`
`225`	`225`
`226`	`226`	`self._max_depth = max_depth`