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