# VITON-HD base / dresscode
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python train_revise_4.py \
 --data_root_dir "./DATA/dresscode"\
 --max_epochs 200\
 --num_sanity_val_steps 1\
 --batch_size 40\
 --logger_freq 1000\
 --valid_epoch_freq 10\
 --save_every_n_epochs 10\
 --precision 16\
 --accum_iter 2 \
 --category "lower_body"\
 --save_name "lower_body_4_from_tryon_upper99"


