diff --git a/README.md b/README.md index c01c4df..de5b8cb 100644 --- a/README.md +++ b/README.md @@ -82,7 +82,7 @@ python -m torch.distributed.launch --nproc_per_node=1 --master_port=48798 main_f --eval --batch_size 16 \ --world_size 1 \ --model vit_large_patch16 \ - --epochs 40 \ + --epochs 50 \ --blr 5e-3 --layer_decay 0.65 \ --weight_decay 0.05 --drop_path 0.2 \ --nb_classes 5 \