From fdb5eb8252cd26c867e89fbc9883df2d14719b29 Mon Sep 17 00:00:00 2001
From: gong1 <b.gong@fz-juelich.de>
Date: Thu, 2 Sep 2021 12:19:45 +0200
Subject: [PATCH] update model_hparam_template for era5 dataset
---
.../hparams/era5/savp/model_hparams_template.json | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/video_prediction_tools/hparams/era5/savp/model_hparams_template.json b/video_prediction_tools/hparams/era5/savp/model_hparams_template.json
index 2275e60f..f36e1c0b 100644
--- a/video_prediction_tools/hparams/era5/savp/model_hparams_template.json
+++ b/video_prediction_tools/hparams/era5/savp/model_hparams_template.json
@@ -1,5 +1,5 @@
{
- "batch_size": 4,
+ "batch_size": 32,
"lr": 0.0002,
"beta1": 0.5,
"beta2": 0.999,
@@ -12,9 +12,11 @@
"gan_feature_cdist_weight": 0.0,
"state_weight": 0.0,
"nz": 16,
- "max_epochs":2,
+ "max_epochs":4,
"context_frames": 12,
- "opt_var": "0"
+ "opt_var": "0",
+ "decay_steps":[3000,9000],
+ "end_lr": 0.00000008
}
--
GitLab