def get_model_test_config():
  """Returns the test config for model."""
  vocab_size = 10
  hidden_dim = 16
  intermediate_dim = 32
  num_heads = 2
  num_layers = 1
  config = compositional.get_model_config()
  config.vocab_size = vocab_size
  config.target_vocab_size = vocab_size
  config.bi_hidden_dim = hidden_dim
  config.l_hidden_dim = hidden_dim
  config.v_hidden_dim = hidden_dim
  config.l_intermediate_dim = intermediate_dim
  config.v_intermediate_dim = intermediate_dim
  config.bi_num_heads = num_heads
  config.l_num_heads = num_heads
  config.v_num_heads = num_heads
  config.decode_num_heads = num_heads
  config.l_num_layers = num_layers
  config.v_num_layers = num_layers
  config.bi_num_layers = num_layers
  config.decode_num_layers = num_layers
  config.max_decode_step = 1

  return config
Returns the test config for model.
language/gscan/xattn_model/test_utils.py