diff --git a/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py b/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py index 7c506cf452ad21263ab7ff7ea90f56291f6e7f17..b9c76c5a7addef60725008f881168f99e9b0f8a4 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py +++ b/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py @@ -22,7 +22,7 @@ from model_modules.video_prediction.layers.BasicConvLSTMCell import BasicConvLST from tensorflow.contrib.training import HParams class VanillaConvLstmVideoPredictionModel(object): - def __init__(self, sequence_length, mode='train', hparams_dict=None): + def __init__(self, mode='train', hparams_dict=None): """ This is class for building convLSTM architecture by using updated hparameters args: