python test.py --model-name="AttentionCluster" --config=./configs/attention_cluster.txt \
                --log-interval=5 --weights=./checkpoints/AttentionCluster_epoch0