From ba0d717ac50fa3544203cb2e1bf0f554e309cf5b Mon Sep 17 00:00:00 2001
From: stadtler1 <s.stadtler@fz-juelich.de>
Date: Fri, 28 Aug 2020 11:19:02 +0200
Subject: [PATCH] a '=' was missing, Bing told me to insert it.
---
.../scripts/generate_transfer_learning_finetune.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/video_prediction_savp/scripts/generate_transfer_learning_finetune.py b/video_prediction_savp/scripts/generate_transfer_learning_finetune.py
index 7b0c462c..1fcbd1cf 100644
--- a/video_prediction_savp/scripts/generate_transfer_learning_finetune.py
+++ b/video_prediction_savp/scripts/generate_transfer_learning_finetune.py
@@ -447,7 +447,7 @@ def main():
#Get prediction values
feed_dict = {input_ph: input_results[name] for name, input_ph in input_phs.items()}
gen_images = sess.run(model.outputs['gen_images'], feed_dict = feed_dict)#return [batchsize,seq_len,lat,lon,channel]
- assert gen_images.shape[1] = sequence_length-1 #The generate images seq_len should be sequence_len -1, since the last one is not used for comparing with groud truth
+ assert gen_images.shape[1] == sequence_length-1 #The generate images seq_len should be sequence_len -1, since the last one is not used for comparing with groud truth
print("gen_images 20200822:",np.array(gen_images).shape)
#Loop in batch size
for i in range(args.batch_size):
--
GitLab