23db031
649c592
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
data:
batch_size: 4
shuffle: True
num_workers: 4
pin_memory: True
train:
optimizer:
type: Adam
args:
lr: 0.001
weight_decay: 0.0001
scheduler:
type: StepLR
args:
step_size: 10
gamma: 0.1
ema:
enabled: true
decay: 0.995
|
