diff --git a/video_prediction_tools/hparams/era5/ours_vae_l1/model_hparams_template.json b/video_prediction_tools/hparams/era5/ours_vae_l1/model_hparams_template.json
index da906729469a5014239b90d66bb97d4508136165..0acefc42a13583d13ac1c263ea44593a6d5e17d0 100644
--- a/video_prediction_tools/hparams/era5/ours_vae_l1/model_hparams_template.json
+++ b/video_prediction_tools/hparams/era5/ours_vae_l1/model_hparams_template.json
@@ -10,6 +10,7 @@
     "video_sn_gan_weight": 0.0,
     "state_weight": 0.0,
     "nz": 32,
+    "max_epochs":2,
     "context_frames":10,
     "sequence_length":20