add config for osnet_ain_x1_0
parent
9397956920
commit
321e479965
|
@ -0,0 +1,36 @@
|
||||||
|
model:
|
||||||
|
name: 'osnet_ain_x1_0'
|
||||||
|
pretrained: True
|
||||||
|
|
||||||
|
data:
|
||||||
|
type: 'image'
|
||||||
|
sources: ['market1501']
|
||||||
|
targets: ['market1501', 'dukemtmcreid']
|
||||||
|
height: 256
|
||||||
|
width: 128
|
||||||
|
combineall: False
|
||||||
|
transforms: ['random_flip', 'color_jitter']
|
||||||
|
save_dir: 'log/osnet_ain_x1_0_market1501_softmax_cosinelr'
|
||||||
|
|
||||||
|
loss:
|
||||||
|
name: 'softmax'
|
||||||
|
softmax:
|
||||||
|
label_smooth: True
|
||||||
|
|
||||||
|
train:
|
||||||
|
optim: 'amsgrad'
|
||||||
|
lr: 0.0015
|
||||||
|
max_epoch: 100
|
||||||
|
batch_size: 64
|
||||||
|
fixbase_epoch: 10
|
||||||
|
open_layers: ['classifier']
|
||||||
|
lr_scheduler: 'cosine'
|
||||||
|
|
||||||
|
test:
|
||||||
|
batch_size: 300
|
||||||
|
dist_metric: 'cosine'
|
||||||
|
normalize_feature: False
|
||||||
|
evaluate: False
|
||||||
|
eval_freq: -1
|
||||||
|
rerank: False
|
||||||
|
visactmap: False
|
Loading…
Reference in New Issue