From 4f2dc0c868ac989c44929014ea9bd22933c0f373 Mon Sep 17 00:00:00 2001
From: "b.gong" <b.gong@fz-juelich.de>
Date: Fri, 15 May 2020 12:16:14 +0200
Subject: [PATCH] update video_prediction/datasets/era5_dataset_v2.py
---
video_prediction/datasets/era5_dataset_v2.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/video_prediction/datasets/era5_dataset_v2.py b/video_prediction/datasets/era5_dataset_v2.py
index ecee4eaa..9ff97051 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
--
GitLab