def rlmb_base_stochastic_discrete_20k_model_steps():
  """Base SD setting with 20k WM steps."""
  hparams = rlmb_base_stochastic_discrete()
  hparams.model_train_steps = 20000
  return hparams
Base SD setting with 20k WM steps.
tensor2tensor/rl/trainer_model_based_params.py