-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
executable file
·46 lines (45 loc) · 1.08 KB
/
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
42
43
44
45
46
### Experiment Setting ###
# ========================
graph_info = False
# graph_info = True
# ========================
data_volume = 100 # 百分比
# ========================
num_epochs = 200
# ========================
# pre_train = False
pre_train = True
# ========================
clf_layer_num = 1 # 測試 1 ~ 3
# 若實驗要嘗試一次使用多種 transform ,請以"減號"分隔 augmentation 方法,Ex: s_aug = "rot-sh"
# ========================
# Spatial augmentation option:
# rot (rotation)
# sh (shear)
# GB (Guassian Blur)
# JM (Joint Mask)
# CM (Channel Mask)
# ------------------------
s_aug = "rot"
# s_aug = "rot"
# s_aug = "None"
# ========================
# Temporal augmentation option:
# re (reverse)
# shu (shuffle)
# RM (Random Mask)
# SU (Speed Up)_bil
# ------------------------
t_aug = "RM"
# t_aug = "None"
# ========================
# augmentation probability, 05 = 50%
aug_p = "05"
if s_aug == "None":
aug_p = "00"
# ========================
if graph_info:
ckpt_name = f"{s_aug}-{aug_p}-{t_aug}"
else:
ckpt_name = f"SGN_STRA_06"
#ckpt_name = "test"