python -u train.py \
--logdir logs \
--pretrained_model checkpoints/pbe_dim5.ckpt \
--base configs/train_dresscode.yaml