class_docstr: Custom pl.Callback for converting model and data to channels_last.
docstr: Undo the changes to pl_module at end of fit, validate, tests, or predict.
code: def teardown(
      self, trainer, pl_module, stage: Optional[str] = None
  ) -> None:
      """Undo the changes to pl_module at end of fit, validate, tests, or predict."""
      if hasattr(pl_module, "on_before_batch_transfer_origin"):
          setattr(
              pl_module,
              "on_before_batch_transfer",
              pl_module.on_before_batch_transfer_origin,
          )
          delattr(pl_module, "on_before_batch_transfer_origin")
      return super().teardown(trainer, pl_module, stage)
  
file_path: BigDL/python/nano/src/bigdl/nano/utils/pytorch/channel_last.py
imports: from typing import Optional
  from types import MethodType
  from logging import warning
  import torch
  import pytorch_lightning
label_desc: Use image node_id 2 for calling the ChannelsLastCallback obj's underlying member method code with example usage: obj.teardown(trainer, pl_module, stage) and returns: super