v1.0

2023-01-31 09:40:20 +00:00
parent a7329b0b22
commit e0b16feecd
1 changed files with 1 additions and 1 deletions
@@ -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 \