test_nonlocal.sh

python -i test.py --model_name="NONLOCAL" --config=./configs/nonlocal.txt \
                --log_interval=1 --weights=./checkpoints/NONLOCAL_epoch0