Merge branch 'bing_issue#021_improve_training_plot_in_train_dummpy_script' of...

Merge branch 'bing_issue#021_improve_training_plot_in_train_dummpy_script' of https://gitlab.version.fz-juelich.de/toar/ambs into bing_issue#021_improve_training_plot_in_train_dummpy_script

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
 
 

video_prediction_savp/hparams/era5/convLSTM/model_hparams.json

 
 {
-    "batch_size": 10,
+    "batch_size": 4,
     "lr": 0.001,
-    "max_epochs":2,
+    "max_epochs":20,
     "context_frames":10,
     "sequence_length":20,
     "loss_fun":"rmse"

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"