code: def __init__(self, config: Config, *args, **kwargs):
      super().__init__(config, *args, **kwargs)
  
      # Head modules
      self.pooler = BertPooler(self.config)
      self.cls = BertOnlyNSPHead(self.config)
  
      # Loss
      self.ce_loss = torch.nn.CrossEntropyLoss(
          ignore_index=self.config.ignore_index
      )
  
file_path: mmf/mmf/models/transformers/heads/itm.py
imports: from dataclasses import dataclass
  from typing import Dict, List, Optional
  import torch
  from mmf.common.registry import registry
  from mmf.models.transformers.base import BaseTransformerHead
label_desc: Use image node_id 1 to create a new ITM object from inherited base classes: BaseTransformerHead with example: obj = ITM(config)