code: def next_frame_sv2p_atari():
      """SV2P model for atari."""
      hparams = next_frame_sv2p()
      hparams.video_num_input_frames = 4
      hparams.video_num_target_frames = 4
      hparams.action_injection = "multiplicative"
      hparams.num_iterations_1st_stage = 12000
      hparams.num_iterations_2nd_stage = 12000
      hparams.anneal_end = 40000
      hparams.latent_loss_multiplier_schedule = (
          "noisy_linear_cosine_decay"
      )
      hparams.latent_loss_multiplier = 1e-3
      hparams.information_capacity = 0.0
      hparams.small_mode = True
      return hparams
  
file_path: tensor2tensor/tensor2tensor/models/video/sv2p_params.py
imports: from __future__ import division
  from __future__ import print_function
  from tensor2tensor.layers import modalities
  from tensor2tensor.models.video import basic_stochastic
  from tensor2tensor.utils import registry
label_desc: Use image node_id 3 for calling a global function with example usage: next_frame_sv2p_atari() and returns: hparams