VITON-Extends / VITON-Extends-Train /scripts /train_VITON-Extends_e2e.sh
NguyenDinhHieu's picture
Add files using upload-large-folder tool
2bd5c7a verified
raw
history blame contribute delete
352 Bytes
python -m torch.distributed.launch --nproc_per_node=8 --master_port=4736 train_VITON-Extends_e2e.py --name VITON-Extends_e2e \
--VITON-Extends_warp_checkpoint 'checkpoints/VITON-Extends_stage1/VITON-Extends_warp_epoch_101.pth' --resize_or_crop None --verbose --tf_log --batchSize 4 --num_gpus 8 --label_nc 14 --launcher pytorch