From 9f296f639ba3d886711d254152ae10ede15f5b06 Mon Sep 17 00:00:00 2001
From: Alex Lee <alexleegk@gmail.com>
Date: Sat, 7 Apr 2018 20:10:16 -0700
Subject: [PATCH] Add hparams of models used in paper (except for sv2p).
---
.../bair/ours_deterministic_l1/model_hparams.json | 14 ++++++++++++++
.../bair/ours_deterministic_l2/model_hparams.json | 14 ++++++++++++++
hparams/bair/ours_gan/model_hparams.json | 13 +++++++++++++
hparams/bair/ours_vae_gan/model_hparams.json | 12 ++++++++++++
hparams/bair/ours_vae_l1/model_hparams.json | 12 ++++++++++++
hparams/bair/ours_vae_l2/model_hparams.json | 12 ++++++++++++
hparams/bair/sna_l1/model_hparams.json | 12 ++++++++++++
hparams/bair/sna_l2/model_hparams.json | 12 ++++++++++++
.../ours_deterministic_l1/model_hparams.json | 14 ++++++++++++++
.../ours_deterministic_l2/model_hparams.json | 14 ++++++++++++++
.../bair_action_free/ours_gan/model_hparams.json | 13 +++++++++++++
.../ours_vae_gan/model_hparams.json | 12 ++++++++++++
.../ours_vae_l1/model_hparams.json | 12 ++++++++++++
.../ours_vae_l2/model_hparams.json | 12 ++++++++++++
.../kth/ours_deterministic_l1/model_hparams.json | 14 ++++++++++++++
.../kth/ours_deterministic_l2/model_hparams.json | 14 ++++++++++++++
hparams/kth/ours_gan/model_hparams.json | 13 +++++++++++++
hparams/kth/ours_vae_gan/model_hparams.json | 12 ++++++++++++
hparams/kth/ours_vae_l1/model_hparams.json | 12 ++++++++++++
19 files changed, 243 insertions(+)
create mode 100644 hparams/bair/ours_deterministic_l1/model_hparams.json
create mode 100644 hparams/bair/ours_deterministic_l2/model_hparams.json
create mode 100644 hparams/bair/ours_gan/model_hparams.json
create mode 100644 hparams/bair/ours_vae_gan/model_hparams.json
create mode 100644 hparams/bair/ours_vae_l1/model_hparams.json
create mode 100644 hparams/bair/ours_vae_l2/model_hparams.json
create mode 100644 hparams/bair/sna_l1/model_hparams.json
create mode 100644 hparams/bair/sna_l2/model_hparams.json
create mode 100644 hparams/bair_action_free/ours_deterministic_l1/model_hparams.json
create mode 100644 hparams/bair_action_free/ours_deterministic_l2/model_hparams.json
create mode 100644 hparams/bair_action_free/ours_gan/model_hparams.json
create mode 100644 hparams/bair_action_free/ours_vae_gan/model_hparams.json
create mode 100644 hparams/bair_action_free/ours_vae_l1/model_hparams.json
create mode 100644 hparams/bair_action_free/ours_vae_l2/model_hparams.json
create mode 100644 hparams/kth/ours_deterministic_l1/model_hparams.json
create mode 100644 hparams/kth/ours_deterministic_l2/model_hparams.json
create mode 100644 hparams/kth/ours_gan/model_hparams.json
create mode 100644 hparams/kth/ours_vae_gan/model_hparams.json
create mode 100644 hparams/kth/ours_vae_l1/model_hparams.json
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 00000000..25fb9307
--- /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 00000000..6d6145e5
--- /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 00000000..8631fa25
--- /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 00000000..5295839d
--- /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 00000000..0536e4f4
--- /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 00000000..1b4fd290
--- /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 00000000..f32b4e7b
--- /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 00000000..992882e5
--- /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 00000000..a499cd54
--- /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 00000000..b961c647
--- /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 00000000..9874d3ee
--- /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 00000000..95c70ceb
--- /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 00000000..827757e1
--- /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 00000000..4fddf0ee
--- /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 00000000..a499cd54
--- /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 00000000..b961c647
--- /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 00000000..9874d3ee
--- /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 00000000..7638a1ca
--- /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 00000000..efef6a00
--- /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
--
GitLab