need memory estimate for window size fig

billbrod · Oct 22, 2020 · 7d938ea · 7d938ea
1 parent 995c43c
commit 7d938ea
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/Snakefile b/Snakefile
@@ -1073,6 +1073,8 @@ rule window_size_figure:
                        '{image_name}_scaling-{scaling}_seed-{seed}_gpu-{gpu}_window_benchmark.txt'))
     params:
         cache_dir = lambda wildcards: op.join(config['DATA_DIR'], 'windows_cache'),
+    resources:
+        mem = get_mem_estimate,
     run:
         import foveated_metamers as met
         import seaborn as sns

diff --git a/prince.json b/prince.json
@@ -27,6 +27,10 @@
         "cpus_per_task": "{resources.cpus_per_task}"
     },
     "cache_windows":
+    {
+        "mem": "{resources.mem}GB"
+    },
+    "window_size_figure":
     {
         "mem": "{resources.mem}GB"
     }

diff --git a/rusty.json b/rusty.json
@@ -28,6 +28,10 @@
         "time": "{params.time}"
     },
     "cache_windows":
+    {
+        "mem": "{resources.mem}GB"
+    },
+    "window_size_figure":
     {
         "mem": "{resources.mem}GB"
     }