def rlmb_base_sv2p_softmax():
  """Base setting with sv2p as world model with softmax."""
  hparams = rlmb_base_sv2p()
  hparams.generative_model_params = "next_frame_sv2p_atari_softmax"
  return hparams
Base setting with sv2p as world model with softmax.
tensor2tensor/rl/trainer_model_based_params.py