python main.py \
    --lr-warm-up  \
    --num-labeled  1000 \
    --num-steps 100000 \
    --warm-up-steps 10000 \
    --batch-size-lab 64 \
    --batch-size-unlab 320 \
    --confidence-mask 0.6 \
    --softmax-temp 0.5 \
    --seed 3 \
    --rot