Increase hidden size

davidnabergoj · Jul 9, 2024 · 9d9b2e6 · 9d9b2e6
1 parent 5a428b8
commit 9d9b2e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/normalizing_flows/neural_networks/convnet.py b/normalizing_flows/neural_networks/convnet.py
@@ -94,7 +94,7 @@ def __init__(self, input_shape, n_outputs: int, kernels: Tuple[int, ...] = None)
 
         self.blocks = nn.ModuleList([reducer] + blocks)
 
-        hidden_size_sqrt: int = 4
+        hidden_size_sqrt: int = 10
         hidden_size = hidden_size_sqrt ** 2
         self.blocks.append(
             ConvModifier(