code: def get_test_cfg():
      vocab_size = 500
      num_token_types = 3
      num_layers = 3
      num_heads = 2
      units = 64
      hidden_size = 96
      hidden_dropout_prob = 0.0
      attention_dropout_prob = 0.0
      cfg = AlbertModel.get_cfg().clone()
      cfg.defrost()
      cfg.MODEL.vocab_size = vocab_size
      cfg.MODEL.num_token_types = num_token_types
      cfg.MODEL.units = units
      cfg.MODEL.hidden_size = hidden_size
      cfg.MODEL.num_heads = num_heads
      cfg.MODEL.num_layers = num_layers
      cfg.MODEL.hidden_dropout_prob = hidden_dropout_prob
      cfg.MODEL.attention_dropout_prob = attention_dropout_prob
      return cfg
  
file_path: gluon-nlp/tests/test_models_albert.py
imports: import pytest
  import numpy
  from numpy.testing import assert_allclose
  import mxnet
  import tempfile
  from gluonnlp.models.albert import AlbertModel, AlbertForMLM, AlbertForPretrain, list_pretrained_albert, get_pretrained_albert
label_desc: Use image node_id 1 for calling a global function with example usage: get_test_cfg() and returns: cfg