class_docstr: Plugin for training with half precision.
  
  Args:
      precision: Whether to use ``torch.float16`` (``'16-true'``) or ``torch.bfloat16`` (``'bf16-true'``).
class_docstr_tok: ["Plugin","for","training","with","half","precision",".","Args",":","precision",":","Whether","to","use","``","torch.float16","``","(","``","'16-true","'","``",")","or","``","torch.bfloat16","``","(","``","'bf16-true","'","``",")","."]
code: def convert_module(self, module: Module) -> Module:
      return module.to(dtype=self._desired_input_dtype)
  
file_path: pytorch-lightning/src/lightning/fabric/plugins/precision/half.py
imports: from typing import Any, ContextManager, Literal
  import torch
  from lightning_utilities.core.apply_func import apply_to_collection
  from torch import Tensor
  from torch.nn import Module
  from typing_extensions import override
  from lightning.fabric.plugins.precision.precision import Precision
  from lightning.fabric.plugins.precision.utils import _convert_fp_tensor, _DtypeContextManager
label_desc: Use image node_id 2 for calling the HalfPrecision obj's underlying member method code with example usage: obj.convert_module(module) and returns: module