From 4f70b33fd260d71ee627a8bac512d9e251ceda56 Mon Sep 17 00:00:00 2001
From: Alex Lee <alexleegk@gmail.com>
Date: Tue, 27 Mar 2018 17:20:56 -0700
Subject: [PATCH] Change batch_size default to 32 in sv2p model.
---
video_prediction/models/sv2p_model.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/video_prediction/models/sv2p_model.py b/video_prediction/models/sv2p_model.py
index 135a2e04..f3205f33 100644
--- a/video_prediction/models/sv2p_model.py
+++ b/video_prediction/models/sv2p_model.py
@@ -610,6 +610,7 @@ class SV2PVideoPredictionModel(VideoPredictionModel):
def get_default_hparams_dict(self):
default_hparams = super(SV2PVideoPredictionModel, self).get_default_hparams_dict()
hparams = dict(
+ batch_size=32,
l1_weight=0.0,
l2_weight=1.0,
kl_weight=1e-3 * 10 * 8, # equivalent to latent_loss_multiplier up to a factor (see below)
--
GitLab