code: def test_summary_parameter_count(max_depth):
      """Test that the summary counts the number of parameters in every submodule."""
      model = UnorderedModel()
      summary = summarize(model, max_depth=max_depth)
      assert summary.param_nums == [
          model.layer2.weight.numel() + model.layer2.bias.numel(),
          model.combine.weight.numel() + model.combine.bias.numel(),
          model.layer1.weight.numel() + model.layer1.bias.numel(),
          0,  # ReLU
          model.unused.weight.numel() + model.unused.bias.numel(),
      ]
  
imports: from collections import OrderedDict
  from typing import Any
  import pytest
  import torch
  import torch.nn
  from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_0
  from lightning.pytorch import LightningModule, Trainer
  from lightning.pytorch.demos.boring_classes import BoringModel
  from lightning.pytorch.utilities.model_summary.model_summary import LEFTOVER_PARAMS_NAME, NOT_APPLICABLE, UNKNOWN_SIZE, ModelSummary, summarize
  from tests_pytorch.helpers.advanced_models import ParityModuleRNN
  from tests_pytorch.helpers.runif import RunIf
label_desc: Use image node_id 7 for calling a global function with example usage: test_summary_parameter_count(max_depth) without return types