code: def fill_hparams(hparams, in_frames, out_frames):
      hparams.video_num_input_frames = in_frames
      hparams.video_num_target_frames = out_frames
      problem = registry.problem("video_stochastic_shapes10k")
      p_hparams = problem.get_hparams(hparams)
      hparams.problem = problem
      hparams.problem_hparams = p_hparams
      hparams.tiny_mode = True
      hparams.reward_prediction = False
      return hparams
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.data_generators import video_generated
  from tensor2tensor.layers import modalities
  from tensor2tensor.utils import registry
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 1 for calling a global function with example usage: fill_hparams(hparams, in_frames, out_frames) and returns: hparams