def rlmb_base_stochastic_discrete_75k_model_steps():
  """Base setting with stochastic discrete model with 75k WM steps."""
  hparams = rlmb_base_stochastic_discrete()
  hparams.model_train_steps = 15000 * 5
  return hparams
Base setting with stochastic discrete model with 75k WM steps.
tensor2tensor/rl/trainer_model_based_params.py