v1.0
This commit is contained in:
@@ -82,7 +82,7 @@ python -m torch.distributed.launch --nproc_per_node=1 --master_port=48798 main_f
|
|||||||
--eval --batch_size 16 \
|
--eval --batch_size 16 \
|
||||||
--world_size 1 \
|
--world_size 1 \
|
||||||
--model vit_large_patch16 \
|
--model vit_large_patch16 \
|
||||||
--epochs 40 \
|
--epochs 50 \
|
||||||
--blr 5e-3 --layer_decay 0.65 \
|
--blr 5e-3 --layer_decay 0.65 \
|
||||||
--weight_decay 0.05 --drop_path 0.2 \
|
--weight_decay 0.05 --drop_path 0.2 \
|
||||||
--nb_classes 5 \
|
--nb_classes 5 \
|
||||||
|
|||||||
Reference in New Issue
Block a user