perf: hopefully this model configuration is faster to train

zimmerman-team · Nov 18, 2021 · 1a2f5d2 · 1a2f5d2
1 parent cd1fb6e
commit 1a2f5d2
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/model_config/dspn_deepnarrow.yaml b/model_config/dspn_deepnarrow.yaml
@@ -0,0 +1,13 @@
+experiment_name: dspn_first
+layers_width: 64
+latent_dim: 16
+activation_function: "ELU"
+depth: 3
+weight_decay: 1.0e-4
+max_epochs: 100
+batch_size: 1024
+divide_output_layer: True
+latent_l1_norm: 10.0
+max_set_size: 50
+dspn_iter: 10
+dspn_lr: 800