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 8e96727e95f761efc170365abd4e8af89696c168..770f9ff516a630ff031b94bb2c8a2b41c1686eec 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 @@ -11,5 +11,5 @@ "state_weight": 0.0, "nz": 32, "max_epochs":2, - "context_frames":10 + "context_frames":12 } diff --git a/video_prediction_tools/hparams/era5/vae/model_hparams_template.json b/video_prediction_tools/hparams/era5/vae/model_hparams_template.json index 341ce4ab333a1c264d682a90bb3019e3f2730225..1306627e24bec0888600fb88fcaa937e5f01dbd7 100644 --- a/video_prediction_tools/hparams/era5/vae/model_hparams_template.json +++ b/video_prediction_tools/hparams/era5/vae/model_hparams_template.json @@ -4,7 +4,7 @@ "lr": 0.001, "nz":16, "max_epochs":2, - "context_frames":10, + "context_frames":12, "weight_recon":1, "loss_fun": "rmse", "shuffle_on_val": true