Mercurial > repos > imgteam > superdsm

comparison run-superdsm.py @ 2:244f67290d28 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tree/master/tools/superdsm/ commit 2286a6c9da88596349ed9d967c51541409c0a7bf
author imgteam
date Mon, 13 Nov 2023 22:12:35 +0000
parents 700ae37e5c69
children 7fd8dba15bd3
comparison
equal deleted inserted replaced
1:700ae37e5c69 2:244f67290d28
10 import imghdr 10 import imghdr
11 import os 11 import os
12 import pathlib 12 import pathlib
13 import shutil 13 import shutil
14 import tempfile 14 import tempfile
15
16 import ray
17 import superdsm.automation
18 import superdsm.io
19 import superdsm.render
20 15
21 16
22 hyperparameters = [ 17 hyperparameters = [
23 ('AF_scale', float), 18 ('AF_scale', float),
24 ('c2f_region_analysis/min_atom_radius', float), 19 ('c2f_region_analysis/min_atom_radius', float),
69 num_threads_per_process = 1 64 num_threads_per_process = 1
70 num_processes = 1 65 num_processes = 1
71 66
72 os.environ['MKL_NUM_THREADS'] = str(num_threads_per_process) 67 os.environ['MKL_NUM_THREADS'] = str(num_threads_per_process)
73 os.environ['OPENBLAS_NUM_THREADS'] = str(num_threads_per_process) 68 os.environ['OPENBLAS_NUM_THREADS'] = str(num_threads_per_process)
74 os.environ['MKL_DEBUG_CPU_TYPE'] = '5' 69
70 import ray
71 import superdsm.automation
72 import superdsm.io
73 import superdsm.render
75 74
76 ray.init(num_cpus=num_processes, log_to_driver=True) 75 ray.init(num_cpus=num_processes, log_to_driver=True)
77 76
78 with tempfile.TemporaryDirectory() as tmpdirname: 77 with tempfile.TemporaryDirectory() as tmpdirname:
79 tmpdir = pathlib.Path(tmpdirname) 78 tmpdir = pathlib.Path(tmpdirname)