From 1293606cbc16ee833624c283a119db008b3588cc Mon Sep 17 00:00:00 2001 From: David Nabergoj Date: Fri, 13 Oct 2023 08:17:35 +0200 Subject: [PATCH] Fix default hidden layers in conditioners --- .../finite/autoregressive/conditioner_transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/normalizing_flows/bijections/finite/autoregressive/conditioner_transforms.py b/normalizing_flows/bijections/finite/autoregressive/conditioner_transforms.py index 53ca5ce..1bf15e5 100644 --- 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: