python test.py --model_name="STNET" --config=./configs/stnet.txt \
                --log_interval=10 --weights=./checkpoints/STNET_epoch0