From 39d9ffbc323aabac824a6614c1a71ec76f2b8712 Mon Sep 17 00:00:00 2001
From: Bing Gong <b.gong@fz-juelich.de>
Date: Thu, 24 Sep 2020 09:23:36 +0000
Subject: [PATCH] correct the train_iterator function
---
video_prediction_savp/main_scripts/main_train_models.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/video_prediction_savp/main_scripts/main_train_models.py b/video_prediction_savp/main_scripts/main_train_models.py
index 0417a365..d3168817 100644
--- a/video_prediction_savp/main_scripts/main_train_models.py
+++ b/video_prediction_savp/main_scripts/main_train_models.py
@@ -144,10 +144,10 @@ def make_dataset_iterator(train_dataset, val_dataset, batch_size ):
val_tf_dataset = val_dataset.make_dataset_v2(batch_size)
val_iterator = val_tf_dataset.make_one_shot_iterator()
val_handle = val_iterator.string_handle()
- #iterator = tf.data.Iterator.from_string_handle(
- # train_handle, train_tf_dataset.output_types, train_tf_dataset.output_shapes)
+ iterator = tf.data.Iterator.from_string_handle(
+ train_handle, train_tf_dataset.output_types, train_tf_dataset.output_shapes)
inputs = train_iterator.get_next()
- val = val_iterator.get_next()
+
return inputs,train_handle, val_handle
--
GitLab