def neural_assistant_tiny_ds():
  """HParams for tiny neural_assistant model with distant supervision loss."""
  hparams = neural_assistant_tiny()
  hparams.kb_loss_weight = 0.2
  return hparams
HParams for tiny neural_assistant model with distant supervision loss.
tensor2tensor/models/neural_assistant.py