def test_problem_hparams(input_vocab_size=None,
                         target_vocab_size=None,
                         model_hparams=None):
  """Problem hparams for testing model bodies."""
  p = TestProblem(input_vocab_size, target_vocab_size)
  return p.get_hparams(model_hparams)
Problem hparams for testing model bodies.
tensor2tensor/data_generators/problem_hparams.py