# VITON-HD base / dresscode bsz=96
CUDA_VISIBLE_DEVICES=0,1,2,3 python train_revise_5.py \
 --data_root_dir "./DATA/VITON-HD"\
 --max_epochs 200\
 --num_sanity_val_steps 1\
 --batch_size 36\
 --logger_freq 1000\
 --valid_epoch_freq 20\
 --save_every_n_epochs 20\
 --precision 16\
 --accum_iter 2 \
 --vitonhd_training \
 --no_strict_load \
 --save_name "viton_tryon_fromScratch_fromtryon179" \
 --resume_path "logs/20250614_viton_tryon_fromScratch_fromDresscodeUpBody99/models/[Train]_[epoch=179]_[train_loss_epoch=0.00000].ckpt"

