python -u test.py \
--base configs/test_dresscode.yaml \
--ckpt checkpoints/dresscode.ckpt \
--ddim 16