36 lines
605 B
YAML
36 lines
605 B
YAML
model:
|
|
name: 'osnet_x0_5'
|
|
pretrained: True
|
|
|
|
data:
|
|
type: 'image'
|
|
sources: ['market1501']
|
|
targets: ['market1501']
|
|
height: 256
|
|
width: 128
|
|
combineall: False
|
|
transforms: ['random_flip']
|
|
save_dir: 'log/osnet_x0_5_market1501_softmax'
|
|
|
|
loss:
|
|
name: 'softmax'
|
|
softmax:
|
|
label_smooth: True
|
|
|
|
train:
|
|
optim: 'amsgrad'
|
|
lr: 0.003
|
|
max_epoch: 180
|
|
batch_size: 128
|
|
fixbase_epoch: 10
|
|
open_layers: ['classifier']
|
|
lr_scheduler: 'single_step'
|
|
stepsize: [80]
|
|
|
|
test:
|
|
batch_size: 300
|
|
dist_metric: 'euclidean'
|
|
normalize_feature: False
|
|
evaluate: False
|
|
eval_freq: -1
|
|
rerank: False |