python train.py --model_name="NONLOCAL" --config=./configs/nonlocal.txt --epoch_num=120 \
                --valid_interval=1 --log_interval=1 \
                --pretrain=./pretrained/ResNet50_pretrained