updates memory estimates for mcmc_plots

billbrod · Apr 13, 2021 · 53b2eef · 53b2eef
1 parent 370c6cb
commit 53b2eef
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/Snakefile b/Snakefile
@@ -1257,6 +1257,8 @@ rule mcmc_plots:
         op.join(config["DATA_DIR"], 'logs', 'mcmc', '{model_name}', 'task-split_comp-{comp}',
                 'task-split_comp-{comp}_mcmc_step-{step_size}_prob-{accept_prob}_depth-{tree_depth}'
                 '_c-{num_chains}_d-{num_draws}_w-{num_warmup}_s-{seed}_{plot_type}_benchmark.txt'),
+    resources:
+        mem = lambda wildcards: {'post-pred-check': 15, 'distribs': 50}.get(wildcards.plot_type, 5)
     run:
         import foveated_metamers as fov
         import arviz as az

diff --git a/rusty.json b/rusty.json
@@ -60,5 +60,9 @@
         "partition": "ccn",
         "mem" : "",
         "time": "6:00:00"
+    },
+    "mcmc_plots":
+    {
+        "mem": "{resources.mem}GB"
     }
 }