Skip to content
Snippets Groups Projects
Commit bd656226 authored by stadtler1's avatar stadtler1
Browse files

added a comment for the loss function

parent e8309fe5
Branches
Tags
No related merge requests found
Pipeline #42593 failed
...@@ -69,6 +69,7 @@ class VanillaConvLstmVideoPredictionModel(BaseVideoPredictionModel): ...@@ -69,6 +69,7 @@ class VanillaConvLstmVideoPredictionModel(BaseVideoPredictionModel):
print("self.x_hat_context_frames,",self.x_hat_context_frames) print("self.x_hat_context_frames,",self.x_hat_context_frames)
#self.context_frames_loss = tf.reduce_mean( #self.context_frames_loss = tf.reduce_mean(
# tf.square(self.x[:, :self.context_frames, :, :, 0] - self.x_hat_context_frames[:, :, :, :, 0])) # 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( 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])) tf.square(self.x[:, self.context_frames:, :, :, 0] - self.x_hat_context_frames[:, (self.context_frames-1):-1, :, :, 0]))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment