code: def next_frame_sv2p_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
docs: SV2P model for atari.