diff --git a/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml b/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml index 2b536e1b99cf9656c8ed141c50e62bfee6bfc762..9ef4beb79db475869414ec3c6e7b9ade3e24b50a 100644 --- 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: