From 07b66e90a17281d1eeb54ed0ab77004f3c0a209a Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Tue, 17 Aug 2021 13:34:27 +0200
Subject: [PATCH] Minor fix in main_train_models.py.
---
video_prediction_tools/main_scripts/main_train_models.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/video_prediction_tools/main_scripts/main_train_models.py b/video_prediction_tools/main_scripts/main_train_models.py
index a23d59cd..e092a855 100644
--- a/video_prediction_tools/main_scripts/main_train_models.py
+++ b/video_prediction_tools/main_scripts/main_train_models.py
@@ -378,7 +378,7 @@ class TrainModel(object):
self.saver_loss = fetch_list[-1]
self.saver_loss_name = "Total loss"
if self.video_model.__class__.__name__ == "ConvLstmGANVideoPredictionModel":
- fetch_list = fetch_list + [inputs, "total_loss", "inputs"]
+ fetch_list = fetch_list + ["inputs", "total_loss"]
self.saver_loss = fetch_list[-1]
self.saver_loss_name = "Total loss"
--
GitLab