diff --git a/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_5.yaml b/ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_x0_5.yaml index 8bacd5ba58b26f8978ef7b67a2684583cf8e9940..aa9591cf32f2de76dfe31b92b1a9de57e83086e4 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 8ba5fdf93d1ba0e5fac7d20d27cb723964ccf10d..837c1ace98cf8741c3b19a951fd6578c207e8404 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 c89cff5b8715467f892eafc163973c73ee8a0345..f9de6f5594dc7a3c37b4b77aec4228ac11d6ff15 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