CUDA_VISIBLE_DEVICES=0 python3 main.py -a resnet50 -b 256 --lr 0.003 --weight-decay 1e-1 --adamw --warmup 8 --opt-level O3 --gradient-accumulation 4 ~/path/to/imagenet/
CUDA_VISIBLE_DEVICES=0 python3 main.py -a resnet50 -b 256 --lr 0.2 --weight-decay 1e-4 --lars --warmup 5 --opt-level O3 ~/path/to/imagenet/
