diff --git a/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py b/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py
index 092cf81a0db1adeb3c3bac65cb3bc92e54e3e4c5..acd08cd8d6a17a602763a6d5a670cd839c61aab4 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py
@@ -256,8 +256,6 @@ class ConvLstmGANVideoPredictionModel(object):
def get_gen_loss(self):
"""
Param:
- num_images: the number of images the generator should produce, which is also the lenght of the real image
- z_dim : the dimension of the noise vector, a scalar
Return the loss of generator given inputs
"""
real_labels = tf.ones_like(self.D_fake)