code: def _create_summary_callback(max_depth: int) -> ModelSummary:
      """Initializes the model summary callback.
      See `ModelSummary reference documentation
      <https://pytorch-lightning.readthedocs.io/en/stable/api/pytorch_lightning.callbacks.ModelSummary.html?highlight=ModelSummary>`.
  
      Args:
          max_depth:
              The maximum depth of layer nesting that the summary will include.
      """
      return ModelSummary(max_depth=max_depth)
  
file_path: lightly/lightly/embedding/callbacks.py
imports: import os
  from typing import Optional
  from omegaconf import DictConfig
  from pytorch_lightning.callbacks import ModelCheckpoint, ModelSummary
  from lightly.utils.hipify import print_as_warning
label_desc: Use image node_id 3 for calling a global function with example usage: _create_summary_callback(max_depth) and returns: ModelSummary