diff --git a/ScaFFold/cli.py b/ScaFFold/cli.py index 840a7e6..6ec5f1d 100644 --- a/ScaFFold/cli.py +++ b/ScaFFold/cli.py @@ -193,9 +193,7 @@ def main(): # Recalculate unet_layers to capture any CLI overrides combined_config["unet_layers"] = ( - combined_config["problem_scale"] - - combined_config["unet_bottleneck_dim"] - + 1 + combined_config["problem_scale"] - combined_config["unet_bottleneck_dim"] ) # Resolve paths to absolute, matching Config() behavior diff --git a/ScaFFold/utils/config_utils.py b/ScaFFold/utils/config_utils.py index 08cb481..fdd5618 100644 --- a/ScaFFold/utils/config_utils.py +++ b/ScaFFold/utils/config_utils.py @@ -45,7 +45,7 @@ def __init__(self, config_dict): ) self.problem_scale = math.floor(self.problem_scale) self.unet_bottleneck_dim = config_dict["unet_bottleneck_dim"] - self.unet_layers = self.problem_scale - self.unet_bottleneck_dim + 1 + self.unet_layers = self.problem_scale - self.unet_bottleneck_dim self.n_fracts_per_vol = config_dict["n_fracts_per_vol"] self.n_instances_used_per_fractal = config_dict["n_instances_used_per_fractal"] self.scale = 1