From 013cb9562c7870c86bcc1985fb2b4635135cc2f8 Mon Sep 17 00:00:00 2001
From: Alex Lee <alexleegk@gmail.com>
Date: Wed, 16 Jan 2019 16:26:34 -0800
Subject: [PATCH] Update number of iterations of each stage and total number of
iterations to match the paper.
---
video_prediction/models/sv2p_model.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/video_prediction/models/sv2p_model.py b/video_prediction/models/sv2p_model.py
index 6cbcafd7..c2134e41 100644
--- a/video_prediction/models/sv2p_model.py
+++ b/video_prediction/models/sv2p_model.py
@@ -641,8 +641,9 @@ class SV2PVideoPredictionModel(VideoPredictionModel):
multi_latent=False,
latent_std_min=-5.0,
latent_channels=1,
- num_iterations_1st_stage=100000,
- kl_anneal_steps=(200000, 220000),
+ num_iterations_1st_stage=50000,
+ kl_anneal_steps=(100000, 120000),
+ max_steps=200000,
decay_steps=(0, 0), # do not decay the learning rate (doing so produces blurrier images)
)
# Notes on equivalence with reference implementation:
--
GitLab