def rlmb_base_stochastic_discrete_1epoch():
  """Long setting with stochastic discrete model, changed epochs."""
  hparams = rlmb_base_stochastic_discrete()
  hparams.epochs = 1
  hparams.ppo_epochs_num = 3000
  return hparams
Long setting with stochastic discrete model, changed epochs.
tensor2tensor/rl/trainer_model_based_params.py