-
Notifications
You must be signed in to change notification settings - Fork 28
/
cls_mdeq_LARGE.yaml
77 lines (76 loc) · 1.13 KB
/
cls_mdeq_LARGE.yaml
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
GPUS: (0,1,2,3)
LOG_DIR: 'log/'
DATA_DIR: ''
OUTPUT_DIR: 'output/'
WORKERS: 4
PRINT_FREQ: 100
MODEL:
NAME: mdeq
NUM_LAYERS: 10
NUM_CLASSES: 10
NUM_GROUPS: 8
DROPOUT: 0.3
F_THRES: 18
B_THRES: 20
WNORM: true
DOWNSAMPLE_TIMES: 0
EXPANSION_FACTOR: 5
IMAGE_SIZE:
- 32
- 32
EXTRA:
FULL_STAGE:
NUM_MODULES: 1
NUM_BRANCHES: 4
BLOCK: BASIC
HEAD_CHANNELS:
- 14
- 28
- 56
- 112
FINAL_CHANSIZE: 1680
NUM_BLOCKS:
- 1
- 1
- 1
- 1
NUM_CHANNELS:
- 32
- 64
- 128
- 256
FUSE_METHOD: SUM
CUDNN:
BENCHMARK: true
DETERMINISTIC: false
ENABLED: true
DATASET:
DATASET: 'cifar10'
DATA_FORMAT: 'jpg'
ROOT: 'data/cifar10/'
TEST_SET: 'val'
TRAIN_SET: 'train'
AUGMENT: true
TEST:
BATCH_SIZE_PER_GPU: 32
MODEL_FILE: ''
TRAIN:
BATCH_SIZE_PER_GPU: 32
BEGIN_EPOCH: 0
END_EPOCH: 220
RESUME: false
LR_SCHEDULER: 'cosine'
PRETRAIN_STEPS: 20000
LR_FACTOR: 0.1
LR_STEP:
- 30
- 60
- 90
OPTIMIZER: adam
LR: 0.001
WD: 0.0
MOMENTUM: 0.9
NESTEROV: true
SHUFFLE: true
DEBUG:
DEBUG: false