minor fix

2022-07-14 19:21:34 +08:00 · 2022-07-14 19:21:34 +08:00 · a667b488ae
parent ce81a07059
commit a667b488ae
1 changed files with 1 additions and 1 deletions
--- a/configs/vision_transformer/vit-base-p16_ft-64xb64_in1k-384.py
+++ b/configs/vision_transformer/vit-base-p16_ft-64xb64_in1k-384.py
@ -40,4 +40,4 @@ optim_wrapper = dict(clip_grad=dict(max_norm=1.0))
 # NOTE: `auto_scale_lr` is for automatically scaling LR,
 # USER SHOULD NOT CHANGE ITS VALUES.
 # base_batch_size = (64 GPUs) x (64 samples per GPU)
-auto_scale_lr = dict(base_batch_size=256)
+auto_scale_lr = dict(base_batch_size=4096)