try using 5 dim

train_lstm_weighted.py

@@ -117,7 +117,7 @@ def build_input(df, seq_norm, p_chanels):
     if seq_norm:
 
         if final_arr.ndim > 2:
-            n, s, f = final_arr.shape
+            n, s, f, h, w = final_arr.shape
             for index, seq in enumerate(final_arr):
                 
                 for i in range(f):