class_docstr: Hooks to be used for data related stuff.
class_docstr_tok: ["Hooks","to","be","used","for","data","related","stuff","."]
docstr: Called at the end of fit (train + validate), validate, test, or predict.
  
  Args:
      stage: either ``'fit'``, ``'validate'``, ``'test'``, or ``'predict'``
code: def teardown(self, stage: str) -> None:
      """Called at the end of fit (train + validate), validate, test, or predict.
  
      Args:
          stage: either ``'fit'``, ``'validate'``, ``'test'``, or ``'predict'``
  
      """
  
imports: from typing import Any, Dict, Optional
  import torch
  from torch import Tensor
  from torch.optim.optimizer import Optimizer
  from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_0
  from lightning.pytorch.utilities import move_data_to_device
  from lightning.pytorch.utilities.exceptions import MisconfigurationException
  from lightning.pytorch.utilities.types import EVAL_DATALOADERS, STEP_OUTPUT, TRAIN_DATALOADERS
label_desc: Use image node_id 4 for calling the DataHooks obj's underlying member method code with example usage: obj.teardown(stage) without return types