File size: 280 Bytes
9e14838 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | training:
lr: 0.01
weight-decay: 0.0001
bs: 8
val_bs: 8
optimizer: 'SGD'
scheduler: 'cosinelr'
gamma: 0.1
step-size: 5
augmentation: 'max'
test:
bs: 1
model:
image-size: 224
num-classes: 1
dim: 1280
mlp-dim: 512
num-frames: 16
max-identities: 2
|