From 31033bfe9f8af634717a16a78a96a121dcf5b05d Mon Sep 17 00:00:00 2001 From: Michael <m.langguth@fz-juelich.de> Date: Thu, 20 May 2021 16:42:59 +0200 Subject: [PATCH] Remove sequence_length from initialization argument-list of convLSTM-network. --- .../video_prediction/models/vanilla_convLSTM_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 7c506cf4..b9c76c5a 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: -- GitLab