class_docstr: A wrapper class for a model object which executes forward with values automatically
  cast to fp16
  
  Args:
      model (:class:`torch.nn.Module`): a torch model instance
docstr: Execute forward under the torch amp context
docstr_tok: ["Execute","forward","under","the","torch","amp","context"]
code: def forward(self, *args, **kwargs):
      """
      Execute forward under the torch amp context
      """
      return self.model(*args, **kwargs)
  
imports: from colossalai.utils.device import autocast
  import torch.nn
  from torch import Tensor
  from torch.nn.modules.loss import _Loss
  from torch.optim import Optimizer
  from colossalai.interface import OptimizerWrapper
  from colossalai.legacy.utils import clip_grad_norm_fp32
  from ._grad_scaler import GradScaler
label_desc: Use image node_id 2 for calling the TorchAMPModel obj's underlying member method code with example usage: obj.forward() and returns: self