# VITON-HD base / dresscode
CUDA_VISIBLE_DEVICES=0,1 python train_mask_free.py \
 --data_root_dir "./DATA/VITON-HD"\
 --max_epochs 300\
 --num_sanity_val_steps 0\
 --batch_size 16\
 --logger_freq 1000\
 --valid_epoch_freq 30\
 --save_every_n_epochs 30\
 --precision 16\
 --accum_iter 2


