From 375d91e5c0fa724e607e0a5eee35b3b627ce1cb1 Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Tue, 17 Aug 2021 15:55:38 +0200
Subject: [PATCH] Add print of validation loss to check if bugfix is removed.
---
video_prediction_tools/main_scripts/main_train_models.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/video_prediction_tools/main_scripts/main_train_models.py b/video_prediction_tools/main_scripts/main_train_models.py
index 6ee98cfa..98df062b 100644
--- a/video_prediction_tools/main_scripts/main_train_models.py
+++ b/video_prediction_tools/main_scripts/main_train_models.py
@@ -318,6 +318,7 @@ class TrainModel(object):
self.create_fetches_for_val()
self.val_results = sess.run(self.val_fetches, feed_dict={self.train_handle: val_handle_eval})
val_losses.append(self.val_results[list(self.saver_loss)[0]])
+ print(val_losses)
self.write_to_summary()
self.print_results(step, self.results)
# track iteration time
--
GitLab