From bd6562269e40f6988537faa8dfd919aa7eb65b05 Mon Sep 17 00:00:00 2001
From: stadtler1 <s.stadtler@fz-juelich.de>
Date: Fri, 7 Aug 2020 16:44:28 +0200
Subject: [PATCH] added a comment for the loss function
---
.../video_prediction/models/vanilla_convLSTM_model.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/video_prediction_savp/video_prediction/models/vanilla_convLSTM_model.py b/video_prediction_savp/video_prediction/models/vanilla_convLSTM_model.py
index 7560a225..be978eae 100644
--- a/video_prediction_savp/video_prediction/models/vanilla_convLSTM_model.py
+++ b/video_prediction_savp/video_prediction/models/vanilla_convLSTM_model.py
@@ -69,6 +69,7 @@ class VanillaConvLstmVideoPredictionModel(BaseVideoPredictionModel):
print("self.x_hat_context_frames,",self.x_hat_context_frames)
#self.context_frames_loss = tf.reduce_mean(
# tf.square(self.x[:, :self.context_frames, :, :, 0] - self.x_hat_context_frames[:, :, :, :, 0]))
+ # This is the loss function (RMSE):
self.total_loss = tf.reduce_mean(
tf.square(self.x[:, self.context_frames:, :, :, 0] - self.x_hat_context_frames[:, (self.context_frames-1):-1, :, :, 0]))
--
GitLab