From 4f5e3a32538ae90b87b17258c52e078f78783435 Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Tue, 15 Jun 2021 17:25:07 +0200
Subject: [PATCH] Also change defaults for context_frames in ours_gan- and
ours_vae-model.
---
.../hparams/era5/ours_vae_l1/model_hparams_template.json | 2 +-
.../hparams/era5/vae/model_hparams_template.json | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
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 8e96727e..770f9ff5 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 341ce4ab..1306627e 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
--
GitLab