fix: dspn inner learning rate to conf

zimmerman-team · Nov 8, 2021 · 6cbf15b · 6cbf15b
1 parent a0b83e0
commit 6cbf15b
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/model_config/dspn_budget_normal.yaml b/model_config/dspn_budget_normal.yaml
@@ -11,3 +11,4 @@ divide_output_layer: True
 latent_l1_norm: 10.0
 max_set_size: 200
 dspn_iter: 4
+dspn_lr: 800
diff --git a/model_config/dspn_result_normal.yaml b/model_config/dspn_result_normal.yaml
@@ -10,4 +10,5 @@ batch_size: 1024
 divide_output_layer: True
 latent_l1_norm: 10.0
 max_set_size: 200
-dspn_iter: 4
+dspn_iter: 4
+dspn_lr: 800
diff --git a/model_config/dspn_result_short.yaml b/model_config/dspn_result_short.yaml
@@ -11,3 +11,4 @@ divide_output_layer: True
 latent_l1_norm: 10.0
 max_set_size: 200
 dspn_iter: 4
+dspn_lr: 800
diff --git a/models/dspn_autoencoder.py b/models/dspn_autoencoder.py
@@ -112,7 +112,7 @@ def __init__(self, **kwargs):
             self.max_set_size,
             kwargs['item_dim'],
             kwargs['dspn_iter'], # number of iteration on the decoder # FIXME: to conf
-            800, # inner learning rate # FIXME: to conf
+            kwargs['dspn_lr'], # inner learning rate # FIXME: to conf
         )
 
     def _make_target(self,loaded_set):