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