diff --git a/workflow/rules/0_common.smk b/workflow/rules/0_common.smk index dfeb7ac..c64ff4e 100644 --- a/workflow/rules/0_common.smk +++ b/workflow/rules/0_common.smk @@ -23,7 +23,7 @@ def make_target_configs(config): def mem_mb_for_samples(wildcards) -> int: - bytes_per_double = 8 + bytes_per_double = 64 num_chains = config["sample"]["chains"] num_iters = config["sample"]["iter_sampling"] target_config = f"{wildcards.target}/{wildcards.target_config}" diff --git a/workflow/rules/2_compute_target_expectations.smk b/workflow/rules/2_compute_target_expectations.smk index 5a13fff..2bb86e2 100644 --- a/workflow/rules/2_compute_target_expectations.smk +++ b/workflow/rules/2_compute_target_expectations.smk @@ -1,5 +1,5 @@ def mem_mb_for_target_expectations(wildcards) -> int: - bytes_per_double = 8 + bytes_per_double = 64 scale_factor = 10 batch_size = 1_000_000 N = target_configs[wildcards.target_config][1]["N"]