"docs/en_US/contents.rst" did not exist on "598d8de2a9ac9579dea962d22d428b14557da38e"
train.sh 240 Bytes
Newer Older
mashun1's avatar
mashun1 committed
1
2
3
4
CUDA_VISIBLE_DEVICES=5,6,7 python -m torch.distributed.launch --nproc_per_node=3 \
    --master_port=26662 train_scripts/train_controlnet.py \
    configs/pixart_app_config/PixArt_xl2_img1024_controlHed_Half.py \
    --work-dir output/debug