code: def pred_batch(self, obs, target_change_times):
      batch_obs = np.expand_dims(obs, axis=0).astype("float32")
      if target_change_times == 0:
          action = self.stage0_model.pred_batch(batch_obs)
      else:
          action = self.stage1_model.pred_batch(batch_obs)
      return action
  
file_path: PARL/examples/NeurIPS2019-Learn-to-Move-Challenge/final_submit/submit_model.py
imports: import numpy
  from parl import layers
  from mlp_model import ActorModel, CriticModel
  from paddle import fluid
  from parl.utils import logger
label_desc: Use image node_id 2 for calling the SubmitModel obj's underlying member method code with example usage: obj.pred_batch(obs, target_change_times) and returns: action