class_docstr: An abstract class for mixed precision training.
code: def configure(
      self,
      model: nn.Module,
      optimizer: Optional[Optimizer] = None,
      criterion: Optional[Callable] = None,
  ) -> Tuple[nn.Module, OptimizerWrapper, Callable]:
      # TODO: implement this method
      pass
  
file_path: colossalai/colossalai/booster/mixed_precision/mixed_precision_base.py
imports: from abc import ABC, abstractmethod
  from typing import Callable, Optional, Tuple
  import torch.nn
  from torch.optim import Optimizer
  from colossalai.interface import OptimizerWrapper
label_desc: Use image node_id 1 for calling the MixedPrecision obj's underlying member method code with example usage: obj.configure(model, optimizer, criterion) without return types