diff --git a/video_prediction/datasets/era5_dataset_v2.py b/video_prediction/datasets/era5_dataset_v2.py
index ecee4eaadfa298fc373958d5ad7673249d075d28..9ff970517c18155daea43c4646d8c750a3c5a509 100644
--- a/video_prediction/datasets/era5_dataset_v2.py
+++ b/video_prediction/datasets/era5_dataset_v2.py
@@ -85,7 +85,7 @@ class ERA5Dataset_v2(VarLenFeatureVideoDataset):
             # images = tf.decode_raw(parsed_features["images/encoded"],tf.int32)
 
             # images = seq
-            images = tf.reshape(seq, [20, 64, 64, 3], name = "reshape_new")
+            images = tf.reshape(seq, [20, 128, 160, 3], name = "reshape_new")
             print("IMAGES", images)
             seqs["images"] = images
             return seqs