model-code / clean /video /mintime /config /slowfast.yaml
deepsafe's picture
Add stripped inference-only model code mirror
9e14838 verified
Raw History Blame Contribute Delete
351 Bytes
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