def rlmb_base_stochastic_discrete_50k():
  """Base setting with stochastic discrete model with 50k steps."""
  hparams = rlmb_base_stochastic_discrete()
  hparams.num_real_env_frames = 48000
  return hparams
Base setting with stochastic discrete model with 50k steps.
tensor2tensor/rl/trainer_model_based_params.py