From 001cdb0955c6f4fba2c27cb42a0ebaeb745557f1 Mon Sep 17 00:00:00 2001 From: Yang Nie Date: Wed, 8 Mar 2023 23:18:26 +0800 Subject: [PATCH] update MobileViTv3-v2 configs --- ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_5.yaml | 5 ++--- ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_75.yaml | 5 ++--- ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x1_0.yaml | 5 ++--- 3 files changed, 6 insertions(+), 9 deletions(-) diff --git a/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_5.yaml b/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_5.yaml index 8bacd5ba..aa9591cf 100644 --- a/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_5.yaml +++ b/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_5.yaml @@ -48,14 +48,13 @@ Optimizer: beta2: 0.999 epsilon: 1e-8 weight_decay: 0.05 - no_weight_decay_name: .bias norm one_dim_param_no_weight_decay: True lr: # for 8 cards name: Cosine learning_rate: 0.002 eta_min: 0.0002 - warmup_epoch: 20 # 20000 iterations + warmup_epoch: 16 # 20000 iterations warmup_start_lr: 1e-6 # by_epoch: True clip_norm: 10 @@ -107,7 +106,7 @@ DataLoader: name: DistributedBatchSampler batch_size: 128 drop_last: False - shuffle: False + shuffle: True loader: num_workers: 4 use_shared_memory: True diff --git a/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_75.yaml b/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_75.yaml index 8ba5fdf9..837c1ace 100644 --- a/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_75.yaml +++ b/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_75.yaml @@ -48,14 +48,13 @@ Optimizer: beta2: 0.999 epsilon: 1e-8 weight_decay: 0.05 - no_weight_decay_name: .bias norm one_dim_param_no_weight_decay: True lr: # for 8 cards name: Cosine learning_rate: 0.002 eta_min: 0.0002 - warmup_epoch: 20 # 20000 iterations + warmup_epoch: 16 # 20000 iterations warmup_start_lr: 1e-6 # by_epoch: True clip_norm: 10 @@ -107,7 +106,7 @@ DataLoader: name: DistributedBatchSampler batch_size: 128 drop_last: False - shuffle: False + shuffle: True loader: num_workers: 4 use_shared_memory: True diff --git a/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x1_0.yaml b/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x1_0.yaml index c89cff5b..f9de6f55 100644 --- a/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x1_0.yaml +++ b/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x1_0.yaml @@ -48,14 +48,13 @@ Optimizer: beta2: 0.999 epsilon: 1e-8 weight_decay: 0.05 - no_weight_decay_name: .bias norm one_dim_param_no_weight_decay: True lr: # for 8 cards name: Cosine learning_rate: 0.002 eta_min: 0.0002 - warmup_epoch: 20 # 20000 iterations + warmup_epoch: 16 # 20000 iterations warmup_start_lr: 1e-6 # by_epoch: True clip_norm: 10 @@ -107,7 +106,7 @@ DataLoader: name: DistributedBatchSampler batch_size: 128 drop_last: False - shuffle: False + shuffle: True loader: num_workers: 4 use_shared_memory: True -- GitLab