diff --git a/video_prediction_savp/HPC_scripts/train_movingmnist.sh b/video_prediction_savp/HPC_scripts/train_movingmnist.sh
index 9343ef85e4b003f5627fbe2b4212e6e29b5cf426..ee864b5ab991eb2be4bf471fca70a075ec7868b0 100755
--- a/video_prediction_savp/HPC_scripts/train_movingmnist.sh
+++ b/video_prediction_savp/HPC_scripts/train_movingmnist.sh
@@ -10,7 +10,7 @@
 #SBATCH --gres=gpu:1
 #SBATCH --partition=gpus
 #SBATCH --mail-type=ALL
-#SBATCH --mail-user=s.stadtler@fz-juelich.de
+#SBATCH --mail-user=b.gong@fz-juelich.de
 ##jutil env activate -p cjjsc42
 
 
diff --git a/video_prediction_savp/hparams/era5/convLSTM/model_hparams.json b/video_prediction_savp/hparams/era5/convLSTM/model_hparams.json
index d4942bea2ab5d6af424844b74d3769ccf699502f..fde951edd2e6b41965fbdce6ce831c1e154cbd0e 100644
--- a/video_prediction_savp/hparams/era5/convLSTM/model_hparams.json
+++ b/video_prediction_savp/hparams/era5/convLSTM/model_hparams.json
@@ -1,8 +1,8 @@
 
 {
-    "batch_size": 10,
+    "batch_size": 4,
     "lr": 0.001,
-    "max_epochs":2,
+    "max_epochs":20,
     "context_frames":10,
     "sequence_length":20,
     "loss_fun":"rmse"
diff --git a/video_prediction_savp/hparams/moving_mnist/convLSTM/model_hparams.json b/video_prediction_savp/hparams/moving_mnist/convLSTM/model_hparams.json
index b07caa0f35f54c1d5007e9fbc6802fe24f1adac0..b59f6cb2ee96162b2eb6014d7ca6bd37f54d4218 100644
--- a/video_prediction_savp/hparams/moving_mnist/convLSTM/model_hparams.json
+++ b/video_prediction_savp/hparams/moving_mnist/convLSTM/model_hparams.json
@@ -2,7 +2,7 @@
 {
     "batch_size": 10,
     "lr": 0.001,
-    "max_epochs":2,
+    "max_epochs":20,
     "context_frames":10,
     "sequence_length":20,
     "loss_fun":"cross_entropy"