def base_nmt_gnmt_lstm(hparams):
  """Base hparams for LSTM-based GNMT models."""
  ## Adjust attention.
  hparams.output_attention = False

  ## Encoder.
  hparams.add_hparam("encoder_type", U.ENC_GNMT)
  hparams.add_hparam("enc_num_layers", 3)
  hparams.add_hparam("enc_num_residual_layers", 1)

  ## Decoder.
  hparams.add_hparam("dec_num_layers", 3)
  hparams.add_hparam("dec_num_residual_layers", 2)

  return hparams
Base hparams for LSTM-based GNMT models.
language/labs/consistent_zero_shot_nmt/models/basic.py