code: def forward(self, preds, target):
      with LightwoodAutocast():
          cat_labels = target.max(1).indices
          return self.cross_entropy_loss(preds, cat_labels)
  
imports: import torch
  from lightwood.helpers.torch import LightwoodAutocast
label_desc: Use image node_id 2 for calling the TransformCrossEntropyLoss obj's underlying member method code with example usage: obj.forward(preds, target) and returns: self