From a667b488aec2a03b9324a6d44ea2428ab42df299 Mon Sep 17 00:00:00 2001 From: yingfhu Date: Thu, 14 Jul 2022 19:21:34 +0800 Subject: [PATCH] minor fix --- configs/vision_transformer/vit-base-p16_ft-64xb64_in1k-384.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/vision_transformer/vit-base-p16_ft-64xb64_in1k-384.py b/configs/vision_transformer/vit-base-p16_ft-64xb64_in1k-384.py index a4cd4c92..2dae63da 100644 --- 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)