Fix default hidden layers in conditioners

davidnabergoj · Oct 13, 2023 · 1293606 · 1293606
1 parent b1f9ea8
commit 1293606
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/normalizing_flows/bijections/finite/autoregressive/conditioner_transforms.py b/normalizing_flows/bijections/finite/autoregressive/conditioner_transforms.py
@@ -82,7 +82,7 @@ def __init__(self,
         )
 
         if n_hidden is None:
-            n_hidden = int(3 * math.log10(self.n_input_event_dims))
+            n_hidden = max(int(3 * math.log10(self.n_input_event_dims)), 4)
 
         # Set conditional dimension values
         ms = [
@@ -151,7 +151,7 @@ def __init__(self,
         )
 
         if n_hidden is None:
-            n_hidden = int(3 * math.log10(self.n_input_event_dims))
+            n_hidden = max(int(3 * math.log10(self.n_input_event_dims)), 4)
 
         # If context given, concatenate it to transform input
         if context_shape is not None: