diff --git a/Snakefile b/Snakefile index 0758ecb..f7bf511 100644 --- a/Snakefile +++ b/Snakefile @@ -576,9 +576,9 @@ rule create_metamers: resources: gpu = lambda wildcards: int(wildcards.gpu), mem = get_mem_estimate, - partition = lambda wildcards: {0: 'gen'}.get(int(wildcards.gpu), 'gpu') params: cache_dir = lambda wildcards: op.join(config['DATA_DIR'], 'windows_cache'), + partition = lambda wildcards: {0: 'gen'}.get(int(wildcards.gpu), 'gpu') run: import foveated_metamers as met import contextlib diff --git a/cluster.json b/prince.json similarity index 100% rename from cluster.json rename to prince.json diff --git a/rusty.json b/rusty.json new file mode 100644 index 0000000..9fc3518 --- /dev/null +++ b/rusty.json @@ -0,0 +1,30 @@ +{ + "__default__": + { + "nodes": 1, + "tasks_per_node": 1, + "mem": "48GB", + "time": "140:00:00", + "job_name": "{rule}.{wildcards}", + "cpus_per_task": 1, + "output": "{log}-%j", + "error": "{log}-%j", + "partition": "gen" + }, + "create_metamers": + { + "gres": "gpu:v100-32gb:{resources.gpu}", + "mem": "{resources.mem}GB", + "partition": "{params.partition}" + }, + "continue_metamers": + { + "gres": "gpu:v100-32gb:{resources.gpu}", + "mem": "{resources.mem}GB", + "partition": "{params.partition}" + }, + "cache_windows": + { + "mem": "{resources.mem}GB" + } +}