Skip to content
Snippets Groups Projects
Commit 4f70b33f authored by Alex Lee's avatar Alex Lee
Browse files

Change batch_size default to 32 in sv2p model.

parent 05fb21ab
No related branches found
No related tags found
No related merge requests found
...@@ -610,6 +610,7 @@ class SV2PVideoPredictionModel(VideoPredictionModel): ...@@ -610,6 +610,7 @@ class SV2PVideoPredictionModel(VideoPredictionModel):
def get_default_hparams_dict(self): def get_default_hparams_dict(self):
default_hparams = super(SV2PVideoPredictionModel, self).get_default_hparams_dict() default_hparams = super(SV2PVideoPredictionModel, self).get_default_hparams_dict()
hparams = dict( hparams = dict(
batch_size=32,
l1_weight=0.0, l1_weight=0.0,
l2_weight=1.0, l2_weight=1.0,
kl_weight=1e-3 * 10 * 8, # equivalent to latent_loss_multiplier up to a factor (see below) kl_weight=1e-3 * 10 * 8, # equivalent to latent_loss_multiplier up to a factor (see below)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment