-
Notifications
You must be signed in to change notification settings - Fork 7
/
config.yml
43 lines (37 loc) · 891 Bytes
/
config.yml
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
LOGGER_NAME: 'log'
SEED: 1546884941.160048
############################ TRAIN ############################
TRAIN:
FIX_TWO: True
############################ TEST ############################
TEST:
BATCH_SIZE: 32
############################ DATA_LOADER ############################
DATA_LOADER:
NUM_WORKERS: 4
SHUFFLE: True
DATA_ROOT: '../dataset/visda2019'
LIST: 'list'
FOLDER: 'clipart'
############################ AUG ############################
AUG:
TEST_CROP: [320, 320]
#TEST_CROP: [288, 288]
RESIZE: [0, 0]
RND_CROP: [0, 0]
V_FLIP: 0.0
H_FLIP: 0.5
ROTATION: 0.0
BRIGHTNESS: 0.0
CONTRAST: 0.0
SATURATION: 0.0
HUE: 0.0
MULTI_CROP_SIZE: 0
MEAN: [0.485, 0.456, 0.406]
STD: [0.229, 0.224, 0.225]
############################ MODEL ############################
MODEL:
CLASS_NUM: 345
NET: 'efficientnet_b5'
IN_DIM: 2048
EMBED_DIM: 1000