code: def __init__(self):
      super().__init__()
      self.loss_fn = nn.CrossEntropyLoss()
  
file_path: colossalai/tests/test_auto_parallel/test_offload/model_utils.py
imports: import torch
  import torch.nn
  from transformers import BertConfig, BertLMHeadModel, GPT2Config, GPT2LMHeadModel
  from tests.components_to_test.registry import non_distributed_component_funcs
label_desc: Use image node_id 1 to create a new LMLoss object from inherited base classes: nn with example: obj = LMLoss()