# VITONHD base
CUDA_VISIBLE_DEVICES=2,3,4 python train_revise_1.py \
 --data_root_dir "./DATA/VITON-HD"\
 --max_epochs 100\
 --num_sanity_val_steps 1\
 --batch_size 36\
 --logger_freq 1000\
 --valid_epoch_freq 5\
 --save_every_n_epochs 10\
 --precision 16


