diff --git a/hparams/bair/ours_deterministic_l1/model_hparams.json b/hparams/bair/ours_deterministic_l1/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..25fb93077aac742cce5f998c3c316ee741032863 --- /dev/null +++ b/hparams/bair/ours_deterministic_l1/model_hparams.json @@ -0,0 +1,14 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 1.0, + "l2_weight": 0.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0001, + "nz": 0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/bair/ours_deterministic_l2/model_hparams.json b/hparams/bair/ours_deterministic_l2/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..6d6145e56aa4cdd9ca6486fec87ea0d2f04fa65a --- /dev/null +++ b/hparams/bair/ours_deterministic_l2/model_hparams.json @@ -0,0 +1,14 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 0.0, + "l2_weight": 1.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0001, + "nz": 0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/bair/ours_gan/model_hparams.json b/hparams/bair/ours_gan/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..8631fa259be44e0b97a98426fa49ce4cd8d7fdd4 --- /dev/null +++ b/hparams/bair/ours_gan/model_hparams.json @@ -0,0 +1,13 @@ +{ + "batch_size": 16, + "lr": 0.0002, + "beta1": 0.5, + "beta2": 0.999, + "l1_weight": 100.0, + "l2_weight": 0.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 1.0, + "state_weight": 0.001, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/bair/ours_vae_gan/model_hparams.json b/hparams/bair/ours_vae_gan/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..5295839d417e206896e4d7cc75faaa7fe391fe2e --- /dev/null +++ b/hparams/bair/ours_vae_gan/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 16, + "lr": 0.0002, + "beta1": 0.5, + "beta2": 0.999, + "l1_weight": 100.0, + "l2_weight": 0.0, + "kl_weight": 0.1, + "video_sn_vae_gan_weight": 1.0, + "video_sn_gan_weight": 1.0, + "state_weight": 0.001 +} \ No newline at end of file diff --git a/hparams/bair/ours_vae_l1/model_hparams.json b/hparams/bair/ours_vae_l1/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..0536e4f4f2641ae39bfedb2f70cc5112848b4407 --- /dev/null +++ b/hparams/bair/ours_vae_l1/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 1.0, + "l2_weight": 0.0, + "kl_weight": 0.001, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0001 +} \ No newline at end of file diff --git a/hparams/bair/ours_vae_l2/model_hparams.json b/hparams/bair/ours_vae_l2/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..1b4fd290b2314bd2b229e34409b12455c44c540e --- /dev/null +++ b/hparams/bair/ours_vae_l2/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 0.0, + "l2_weight": 1.0, + "kl_weight": 0.001, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0001 +} \ No newline at end of file diff --git a/hparams/bair/sna_l1/model_hparams.json b/hparams/bair/sna_l1/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..f32b4e7b64567d23db3c6594bf913e8c657f2b25 --- /dev/null +++ b/hparams/bair/sna_l1/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 1.0, + "l2_weight": 0.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0001 +} \ No newline at end of file diff --git a/hparams/bair/sna_l2/model_hparams.json b/hparams/bair/sna_l2/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..992882e597889f7875f22982b898619d21ceee66 --- /dev/null +++ b/hparams/bair/sna_l2/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 0.0, + "l2_weight": 1.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0001 +} \ No newline at end of file diff --git a/hparams/bair_action_free/ours_deterministic_l1/model_hparams.json b/hparams/bair_action_free/ours_deterministic_l1/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..a499cd54c8f41e1be838c9cf8f6ee9a8f32d2a57 --- /dev/null +++ b/hparams/bair_action_free/ours_deterministic_l1/model_hparams.json @@ -0,0 +1,14 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 1.0, + "l2_weight": 0.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0, + "nz": 0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/bair_action_free/ours_deterministic_l2/model_hparams.json b/hparams/bair_action_free/ours_deterministic_l2/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..b961c6475921b6ef147c7cf8bf98c37202579d61 --- /dev/null +++ b/hparams/bair_action_free/ours_deterministic_l2/model_hparams.json @@ -0,0 +1,14 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 0.0, + "l2_weight": 1.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0, + "nz": 0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/bair_action_free/ours_gan/model_hparams.json b/hparams/bair_action_free/ours_gan/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..9874d3ee4ba9e75fd04aef51289dead1b7c223f3 --- /dev/null +++ b/hparams/bair_action_free/ours_gan/model_hparams.json @@ -0,0 +1,13 @@ +{ + "batch_size": 16, + "lr": 0.0002, + "beta1": 0.5, + "beta2": 0.999, + "l1_weight": 100.0, + "l2_weight": 0.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 1.0, + "state_weight": 0.0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/bair_action_free/ours_vae_gan/model_hparams.json b/hparams/bair_action_free/ours_vae_gan/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..95c70cebb7a4c2e7503950d505f014d7cefe954f --- /dev/null +++ b/hparams/bair_action_free/ours_vae_gan/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 16, + "lr": 0.0002, + "beta1": 0.5, + "beta2": 0.999, + "l1_weight": 100.0, + "l2_weight": 0.0, + "kl_weight": 0.1, + "video_sn_vae_gan_weight": 1.0, + "video_sn_gan_weight": 1.0, + "state_weight": 0.0 +} \ No newline at end of file diff --git a/hparams/bair_action_free/ours_vae_l1/model_hparams.json b/hparams/bair_action_free/ours_vae_l1/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..827757e11b75e720d236417be449b7a301a005ec --- /dev/null +++ b/hparams/bair_action_free/ours_vae_l1/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 1.0, + "l2_weight": 0.0, + "kl_weight": 0.001, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0 +} \ No newline at end of file diff --git a/hparams/bair_action_free/ours_vae_l2/model_hparams.json b/hparams/bair_action_free/ours_vae_l2/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..4fddf0eef1d45dbfa16f098e5e42b12f594132e3 --- /dev/null +++ b/hparams/bair_action_free/ours_vae_l2/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 0.0, + "l2_weight": 1.0, + "kl_weight": 0.001, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0 +} \ No newline at end of file diff --git a/hparams/kth/ours_deterministic_l1/model_hparams.json b/hparams/kth/ours_deterministic_l1/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..a499cd54c8f41e1be838c9cf8f6ee9a8f32d2a57 --- /dev/null +++ b/hparams/kth/ours_deterministic_l1/model_hparams.json @@ -0,0 +1,14 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 1.0, + "l2_weight": 0.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0, + "nz": 0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/kth/ours_deterministic_l2/model_hparams.json b/hparams/kth/ours_deterministic_l2/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..b961c6475921b6ef147c7cf8bf98c37202579d61 --- /dev/null +++ b/hparams/kth/ours_deterministic_l2/model_hparams.json @@ -0,0 +1,14 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 0.0, + "l2_weight": 1.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0, + "nz": 0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/kth/ours_gan/model_hparams.json b/hparams/kth/ours_gan/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..9874d3ee4ba9e75fd04aef51289dead1b7c223f3 --- /dev/null +++ b/hparams/kth/ours_gan/model_hparams.json @@ -0,0 +1,13 @@ +{ + "batch_size": 16, + "lr": 0.0002, + "beta1": 0.5, + "beta2": 0.999, + "l1_weight": 100.0, + "l2_weight": 0.0, + "kl_weight": 0.0, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 1.0, + "state_weight": 0.0, + "e_net": "none" +} \ No newline at end of file diff --git a/hparams/kth/ours_vae_gan/model_hparams.json b/hparams/kth/ours_vae_gan/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..7638a1cac87d470ce9d7f323ccc5502fc2396be3 --- /dev/null +++ b/hparams/kth/ours_vae_gan/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 16, + "lr": 0.0002, + "beta1": 0.5, + "beta2": 0.999, + "l1_weight": 100.0, + "l2_weight": 0.0, + "kl_weight": 0.001, + "video_sn_vae_gan_weight": 1.0, + "video_sn_gan_weight": 1.0, + "state_weight": 0.0 +} \ No newline at end of file diff --git a/hparams/kth/ours_vae_l1/model_hparams.json b/hparams/kth/ours_vae_l1/model_hparams.json new file mode 100644 index 0000000000000000000000000000000000000000..efef6a0038c6d9b70958dd82e5243ea1d6a02a53 --- /dev/null +++ b/hparams/kth/ours_vae_l1/model_hparams.json @@ -0,0 +1,12 @@ +{ + "batch_size": 32, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "l1_weight": 1.0, + "l2_weight": 0.0, + "kl_weight": 1e-05, + "video_sn_vae_gan_weight": 0.0, + "video_sn_gan_weight": 0.0, + "state_weight": 0.0 +} \ No newline at end of file