-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbatch_config.py
41 lines (31 loc) · 1.47 KB
/
batch_config.py
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
32
33
34
35
36
37
38
39
40
41
import os
import sys
from socket import gethostname
NUM_RUNS = 10 # number of independent simulations
# choose network class in NetworkClass.py
variants = {'NetworkIntDriven': 'IntDrivenLog80', 'NetworkExtDriven': 'ExtDrivenLog100', 'NetworkIndep': 'IndepLog80'}
variants |= {'LinearTrack': 'ManyLaps'} # better for fast/slow analysis with top and bottom 20% percentiles of speed
# variants |= {'PlaceFields': 'HighThreshold'} # comment this out only when using NetworkIndep
# DEFINE PATHS (change according to your file system!)
# This is where pickles and results will be saved (potentially several GB).
# Currently, the code checks in which machine I am, and whether my external drives are mounted or not.
# You can comment that out and simply define "pickles_path" and "figures_path" to whatever your like.
host_name = gethostname()
drive_names = ["INTENSO", "TOSHIBA"]
if host_name == "lux39":
drive_parent_path = "/media/eparra"
local_path = "/local"
elif host_name == "etp":
drive_parent_path = "/media/eloy"
local_path = "/c/DATA/Simulations"
else:
sys.exit("host not recognized")
root_path = local_path
for drive_name in drive_names:
drive_path = f"{drive_parent_path}/{drive_name}"
if os.path.exists(drive_path):
root_path = drive_path
break
root_path = f"{root_path}/BDSweeps"
pickles_path = f"{root_path}/pickles" # Path where the pickles will be saved
figures_path = f"{root_path}/figures" # Path where the figures will be saved