File size: 351 Bytes
9e14838 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | training:
lr: 0.001
weight-decay: 0.0001
bs: 32
val_bs: 32
optimizer: 'SGD'
scheduler: 'cosinelr'
gamma: 0.1
step-size: 5
augmentation: 'max' # min/max
test:
bs: 32
model:
image-size: 256
num-classes: 1
num-frames: 16
max-identities: 1
enable-size-emb: False
enable-pos-emb: False
enable-identity-attention: False
|