update configs

anananan116 · Mar 12, 2024 · a1f83f9 · a1f83f9
1 parent cca447a
commit a1f83f9
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/configs/default_all_bound128.yaml b/configs/default_all_bound128.yaml
@@ -0,0 +1,23 @@
+#dataset settings
+dataset: both
+validation_ratio: 0.01
+test_ratio: 0.01
+img_size: 128
+batch_size: 512
+
+VQ-VAE:
+  version: 2
+  experiment_id: 20
+  count_low_usage: True
+  epochs: 400
+  lr: 0.0004
+  beta: 0.25
+  kernel_sizes:
+    - 3
+    - 3
+    - 3
+  res_layers: 3
+  latent_dimension: 128
+  code_book_size: 512
+  early_stop_patience: 20
+  lower_bound_factor: 128
diff --git a/configs/default_all_bound160.yaml b/configs/default_all_bound160.yaml
@@ -0,0 +1,23 @@
+#dataset settings
+dataset: both
+validation_ratio: 0.01
+test_ratio: 0.01
+img_size: 128
+batch_size: 512
+
+VQ-VAE:
+  version: 2
+  experiment_id: 21
+  count_low_usage: True
+  epochs: 400
+  lr: 0.0004
+  beta: 0.25
+  kernel_sizes:
+    - 3
+    - 3
+    - 3
+  res_layers: 3
+  latent_dimension: 128
+  code_book_size: 512
+  early_stop_patience: 20
+  lower_bound_factor: 160