fixed batch size in padding

cikm_inter_dst_predrnn_run_taasss.py

@@ -128,6 +128,7 @@ def schedule_sampling(eta: float, itr: int):
     )
     return eta, real_input_flag
 
+
 def wrapper_train(model: Model):
     eta = args.sampling_start_value
     iterator = get_batcher(args)

cikm_inter_dst_predrnn_run_taasss_utils.py

@@ -42,7 +42,7 @@ def padding_taasss(array: np.ndarray, args) -> np.ndarray:
     to
     (1, 25, 480, 480, 1)
     """
-    zeros = np.zeros((1, 25, args.img_width, args.img_width, 1))
+    zeros = np.zeros((args.batch_size, 25, args.img_width, args.img_width, 1))
     if args.img_width == 512:
         zeros[:, :, 16:496, 16:496, :] = array
     elif args.img_width == 256: