python train.py \
    --root_dir gaofen_aug \
    --num_workers=8 \
    --net_G=segformer \
    --batch_size=8 \
    --pretrained=pretrained/mit_b4.pth \
    --in_channels=4 \
    --lr=1e-5 \
    --eval_steps=1000 \
    --restore_path=work/models/base_transformer_pos_s4_dd8_dedim8/2021-10-13T21:48:29.pkl \
    --last_step=56000