torchrun --nproc_per_node=8 train.py --model mobilenet_v2 --batch-size 128 --lr 0.5 \
--lr-scheduler cosineannealinglr --lr-warmup-epochs 5 --lr-warmup-method linear --auto-augment ta_wide \
--epochs 600 --random-erase 0.1 --label-smoothing 0.1 --mixup-alpha 0.2 --cutmix-alpha 1.0 \
--weight-decay 0.00002 --norm-weight-decay 0.0 --train-crop-size 176 --model-ema --val-resize-size 232 \
--ra-sampler --ra-reps 4

torchrun --nproc_per_node=1 train.py --test-only --prototype --weights MobileNet_V2_Weights.IMAGENET1K_V2 --model mobilenet_v2 -b 1
Acc@1 72.154 Acc@5 90.822