forked from IBM/CryoRL-pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcryoEM_config.py
41 lines (36 loc) · 990 Bytes
/
cryoEM_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
class CryoEMConfig:
# max ctf number allowed
REGRESSION = 0
CLASSIFICATION = 1
TRAIN = 0
VALIDATION = 1
MAX_CTF_VALUE = 20.0
LOW_CTF_THRESH = 6.0
MAX_HOLE_CNT_PER_PATCH = 25.0
MAX_HOLE_CNT_PER_SQUARE= 150.0
MAX_HOLE_CNT_PER_GRID = 500.0
FEATURE_DIM = 17 #20 #13 #23 #16
HISTORY_SIZE = 4
FEATURE_HISTOGRAM_BIN = [0, 4, 6, 8, 10, 999999]
#FEATURE_HISTOGRAM_BIN = [0, 6, 10, 999999]
GAUSS_FILTER_SIZE = 0.0
# SWITCH_GRID_LOSS = 15
# SWITCH_SQUARE_LOSS = 10
# SWITCH_PATCH_LOSS = 5
# SWITCH_HOLE_LOSS = 2
SWITCH_GRID_LOSS = 10
SWITCH_SQUARE_LOSS = 5
SWITCH_PATCH_LOSS = 3
SWITCH_HOLE_LOSS = 2
# Yilai's config
# SWITCH_GRID_LOSS = 8
# SWITCH_SQUARE_LOSS = 6
# SWITCH_PATCH_LOSS = 2
# SWITCH_HOLE_LOSS = 1
Searching_Limit = 120
LOW_CTF_REWARD = 1
HIGH_CTF_REWARD = 0
EVALUATION_TRIALS = 30
# action elimination factor
TRAIN_AE_FACTOR = 7.5
VAL_AE_FACTOR = 1.5