timm

/

vit_base_patch16_224.augreg_in21k_ft_in1k

like 0

Follow

PyTorch Image Models 148

Please, could you share training parameters for this model? I.e. something of the form like

./distributed_train.sh 4 /data/imagenet --model vit_base_patch16_224.augreg_in21k_ft_in1k --sched cosine --epochs 150 --warmup-epochs 5 --lr 0.4 --reprob 0.5 --remode pixel --batch-size 256 --amp -j 4