fix: fix bs lr

2022-06-13 08:26:36 +00:00 · 2022-06-13 08:26:36 +00:00 · 23b0e1388c
parent 1152121f9f
commit 23b0e1388c
1 changed files with 3 additions and 3 deletions
--- a/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml
+++ b/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml
@ -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: