fix: fix bs lr

This commit is contained in:
gaotingquan 2022-06-13 08:26:36 +00:00
parent 1152121f9f
commit 23b0e1388c
No known key found for this signature in database
GPG Key ID: F3EF7F42536A30B7

View File

@ -36,7 +36,7 @@ Optimizer:
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: Cosine
learning_rate: 0.4 learning_rate: 0.08
warmup_epoch: 5 warmup_epoch: 5
regularizer: regularizer:
name: 'L2' name: 'L2'
@ -66,11 +66,11 @@ DataLoader:
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 64 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 4 num_workers: 8
use_shared_memory: True use_shared_memory: True
Eval: Eval: