class_docstr: Token-based (input/output is tokens) encoder-decoder model (e.g. T5 Language model.)
docstr: Customized load.
docstr_tok: ["Customized","load","."]
code: def load_state_dict(self, state_dict, strict=True):
      """Customized load."""
  
      self.encoder_embedding.encoder_embeddingload_state_dict(
          state_dict[self._encoder_embedding_key], strict=strict
      )
      self.decoder_embedding.load_state_dict(
          state_dict[self._decoder_embedding_key], strict=strict
      )
      self.enc_dec_model.load_state_dict(
          state_dict[self._enc_dec_model_key], strict=strict
      )
      self.tokens_head.load_state_dict(
          state_dict[self._tokens_head_key], strict=strict
      )
  
imports: import torch
  from omegaconf import DictConfig
  from nemo.collections.nlp.modules.common.megatron.adapters.parallel_adapters import AdapterName, PromptEncoderAdapterConfig
  from nemo.collections.nlp.modules.common.megatron.hiddens import get_hiddens_module
  from nemo.collections.nlp.modules.common.megatron.language_model import Embedding
  from nemo.collections.nlp.modules.common.megatron.layer_type import LayerType
  from nemo.collections.nlp.modules.common.megatron.megatron_decoders import get_decoder_model
  from nemo.collections.nlp.modules.common.megatron.megatron_encoder_decoder import MegatronTransformerEncoderDecoderModule
  from nemo.collections.nlp.modules.common.megatron.megatron_encoders import get_encoder_model
  from nemo.collections.nlp.modules.common.megatron.module import MegatronModule
  from nemo.collections.nlp.modules.common.megatron.position_embedding import ALiBiRelativePositionEmbedding, KERPLERelativePositionEmbedding, T5RelativePositionEmbedding
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults, build_position_ids, init_method_normal, parallel_lm_logits, scaled_init_method_normal
  from nemo.collections.nlp.modules.common.megatron.vocab_parallel_cross_entropy import vocab_parallel_cross_entropy
  from nemo.core.classes.mixins import adapter_mixins
label_desc: Use image node_id 9 for calling the MegatronTokenLevelEncoderDecoderModule obj's underlying member method code with example usage: obj.load_state_dict(state_dict, strict) without return types