-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
31 lines (22 loc) · 650 Bytes
/
main.py
File metadata and controls
31 lines (22 loc) · 650 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from itertools import product
from pathlib import Path
from multiprocessing import Process, Pool
from decode import run
PARALLEL = False
def main():
n_processes = 15
exps = ['grasp', 'imagine']
bands = [['beta'], ['high_gamma'], ['beta', 'high_gamma']]
ppts = ['p1', 'p2', 'p3', 'p4', 'p5', 'p6', 'p7', 'p8']
jobs = product(ppts, exps, bands)
if PARALLEL:
pool = Pool(processes=n_processes)
for job in jobs:
pool.apply_async(run, args=((job)))
pool.close()
pool.join()
else:
for job in jobs:
run(*job)
if __name__=='__main__':
main()