wish44165 commited on
Commit
e677d81
·
1 Parent(s): fb1d170

Move config and model to root for download stats

Browse files
Files changed (2) hide show
  1. config.yaml +189 -0
  2. model_0016.pth +3 -0
config.yaml ADDED
@@ -0,0 +1,189 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ NUM_INSTANCE: 16
4
+ NUM_WORKERS: 8
5
+ SAMPLER_TRAIN: BalancedIdentitySampler
6
+ SET_WEIGHT: []
7
+ DATASETS:
8
+ COMBINEALL: false
9
+ NAMES:
10
+ - MOT20
11
+ TESTS:
12
+ - MOT20
13
+ INPUT:
14
+ AFFINE:
15
+ ENABLED: false
16
+ AUGMIX:
17
+ ENABLED: false
18
+ PROB: 0.0
19
+ AUTOAUG:
20
+ ENABLED: true
21
+ PROB: 0.1
22
+ CJ:
23
+ BRIGHTNESS: 0.15
24
+ CONTRAST: 0.15
25
+ ENABLED: false
26
+ HUE: 0.1
27
+ PROB: 0.5
28
+ SATURATION: 0.1
29
+ CROP:
30
+ ENABLED: false
31
+ RATIO:
32
+ - 0.75
33
+ - 1.3333333333333333
34
+ SCALE:
35
+ - 0.16
36
+ - 1
37
+ SIZE:
38
+ - 224
39
+ - 224
40
+ FLIP:
41
+ ENABLED: true
42
+ PROB: 0.5
43
+ PADDING:
44
+ ENABLED: true
45
+ MODE: constant
46
+ SIZE: 10
47
+ REA:
48
+ ENABLED: true
49
+ PROB: 0.5
50
+ VALUE:
51
+ - 123.675
52
+ - 116.28
53
+ - 103.53
54
+ RPT:
55
+ ENABLED: false
56
+ PROB: 0.5
57
+ SIZE_TEST:
58
+ - 256
59
+ - 256
60
+ SIZE_TRAIN:
61
+ - 256
62
+ - 256
63
+ KD:
64
+ EMA:
65
+ ENABLED: false
66
+ MOMENTUM: 0.999
67
+ MODEL_CONFIG: []
68
+ MODEL_WEIGHTS: []
69
+ MODEL:
70
+ BACKBONE:
71
+ ATT_DROP_RATE: 0.0
72
+ DEPTH: 50x
73
+ DROP_PATH_RATIO: 0.1
74
+ DROP_RATIO: 0.0
75
+ FEAT_DIM: 2048
76
+ LAST_STRIDE: 1
77
+ NAME: build_resnest_backbone
78
+ NORM: BN
79
+ PRETRAIN: true
80
+ PRETRAIN_PATH: ''
81
+ SIE_COE: 3.0
82
+ STRIDE_SIZE:
83
+ - 16
84
+ - 16
85
+ WITH_IBN: false
86
+ WITH_NL: true
87
+ WITH_SE: false
88
+ DEVICE: cuda:0
89
+ FREEZE_LAYERS:
90
+ - backbone
91
+ HEADS:
92
+ CLS_LAYER: CircleSoftmax
93
+ EMBEDDING_DIM: 0
94
+ MARGIN: 0.35
95
+ NAME: EmbeddingHead
96
+ NECK_FEAT: after
97
+ NORM: BN
98
+ NUM_CLASSES: 4288
99
+ POOL_LAYER: GeneralizedMeanPoolingP
100
+ SCALE: 64
101
+ WITH_BNNECK: true
102
+ LOSSES:
103
+ CE:
104
+ ALPHA: 0.2
105
+ EPSILON: 0.1
106
+ SCALE: 1.0
107
+ CIRCLE:
108
+ GAMMA: 48
109
+ MARGIN: 0.35
110
+ SCALE: 1.0
111
+ COSFACE:
112
+ GAMMA: 128
113
+ MARGIN: 0.25
114
+ SCALE: 1.0
115
+ FL:
116
+ ALPHA: 0.25
117
+ GAMMA: 2
118
+ SCALE: 1.0
119
+ NAME:
120
+ - CrossEntropyLoss
121
+ - CircleLoss
122
+ TRI:
123
+ HARD_MINING: true
124
+ MARGIN: 0.0
125
+ NORM_FEAT: false
126
+ SCALE: 1.0
127
+ META_ARCHITECTURE: Baseline
128
+ PIXEL_MEAN:
129
+ - 123.675
130
+ - 116.28
131
+ - 103.53
132
+ PIXEL_STD:
133
+ - 58.395
134
+ - 57.120000000000005
135
+ - 57.375
136
+ QUEUE_SIZE: 8192
137
+ WEIGHTS: ''
138
+ OUTPUT_DIR: logs/MOT20/sbs_S50
139
+ SOLVER:
140
+ AMP:
141
+ ENABLED: true
142
+ BASE_LR: 0.00035
143
+ BIAS_LR_FACTOR: 1.0
144
+ CHECKPOINT_PERIOD: 1
145
+ CLIP_GRADIENTS:
146
+ CLIP_TYPE: norm
147
+ CLIP_VALUE: 5.0
148
+ ENABLED: false
149
+ NORM_TYPE: 2.0
150
+ DELAY_EPOCHS: 30
151
+ ETA_MIN_LR: 7.0e-07
152
+ FREEZE_ITERS: 1000
153
+ GAMMA: 0.1
154
+ HEADS_LR_FACTOR: 1.0
155
+ IMS_PER_BATCH: 256
156
+ MAX_EPOCH: 120
157
+ MOMENTUM: 0.9
158
+ NESTEROV: false
159
+ OPT: AdamW
160
+ SCHED: CosineAnnealingLR
161
+ STEPS: []
162
+ WARMUP_FACTOR: 0.1
163
+ WARMUP_ITERS: 2000
164
+ WARMUP_METHOD: linear
165
+ WEIGHT_DECAY: 0.0005
166
+ WEIGHT_DECAY_BIAS: 0.0005
167
+ WEIGHT_DECAY_NORM: 0.0005
168
+ TEST:
169
+ AQE:
170
+ ALPHA: 3.0
171
+ ENABLED: false
172
+ QE_K: 5
173
+ QE_TIME: 1
174
+ EVAL_PERIOD: 1000000
175
+ FLIP:
176
+ ENABLED: false
177
+ IMS_PER_BATCH: 512
178
+ METRIC: cosine
179
+ PRECISE_BN:
180
+ DATASET: Market1501
181
+ ENABLED: false
182
+ NUM_ITER: 300
183
+ RERANK:
184
+ ENABLED: false
185
+ K1: 20
186
+ K2: 6
187
+ LAMBDA: 0.3
188
+ ROC:
189
+ ENABLED: false
model_0016.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f6e20a9082eb5fe100e005bd5d6f2bf9fde7d9aeff5575546ff3c530baf6363
3
+ size 411018623