class_docstr: Transformer decoder model.
      
docstr: Customized load.
docstr_tok: ["Customized","load","."]
code: def load_state_dict(self, state_dict, strict=True):
      """Customized load."""
  
      # Encoder.
      if self._model_key in state_dict:
          state_dict_ = state_dict[self._model_key]
      self.model.load_state_dict(state_dict_, strict=strict)
  
imports: import torch
  import torch.nn.functional
  from einops import rearrange, repeat
  from nemo.collections.nlp.modules.common.megatron.module import MegatronModule
  from nemo.collections.nlp.modules.common.megatron.position_embedding import RotaryEmbedding
  from nemo.collections.nlp.modules.common.megatron.transformer import ParallelTransformer
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults, build_attention_mask_3d
label_desc: Use image node_id 6 for calling the MegatronRetrievalTransformerDecoderModule obj's underlying member method code with example usage: obj.load_state_dict(state_dict, strict) without return types