Skip to content

Commit

Permalink
modified batch_size and update_freq
Browse files Browse the repository at this point in the history
modified batch_size per gpu and update_freq in MobileViTv3_S.yaml for training with 4 gpus
  • Loading branch information
flytocc committed Mar 30, 2023
1 parent 4c9a4de commit a1e6128
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions ppcls/configs/ImageNet/MobileViTv3/MobileViTv3_S.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ Global:
image_shape: [3, 256, 256]
save_inference_dir: ./inference
use_dali: False
update_freq: 3 # for 4 gpus

# mixed precision training
AMP:
Expand Down Expand Up @@ -49,9 +50,8 @@ Optimizer:
epsilon: 1e-8
weight_decay: 0.01
lr:
# for 8 cards
name: Cosine
learning_rate: 0.002
learning_rate: 0.002 # for total batch size 384
eta_min: 0.0002
warmup_epoch: 1 # 3000 iterations
warmup_start_lr: 0.0002
Expand Down Expand Up @@ -86,7 +86,7 @@ DataLoader:
scales: [256, 160, 192, 224, 288, 320]
# first_bs: batch size for the first image resolution in the scales list
# divide_factor: to ensure the width and height dimensions can be devided by downsampling multiple
first_bs: 48
first_bs: 32
divided_factor: 32
is_training: True
loader:
Expand Down

0 comments on commit a1e6128

Please sign in to comment.