From 0b1fe7c1ca4b443bd2647a540b91729c017ff88a Mon Sep 17 00:00:00 2001
From: Bing Gong <b.gong@fz-juelich.de>
Date: Fri, 30 Apr 2021 10:12:30 +0000
Subject: [PATCH] Update model_hparams_template.json

---
 .../hparams/era5/ours_vae_l1/model_hparams_template.json         | 1 +
 1 file changed, 1 insertion(+)

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 da906729..0acefc42 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
 
-- 
GitLab