Apppublic

tohid4n/PartCrafter

sourceHugging Facemitupdated 1y agoView on Hugging Face

0likes

train_partcrafter.sh15 linesDownload Raw Back to scripts

1NUM_MACHINES=12NUM_LOCAL_GPUS=83MACHINE_RANK=04 5export WANDB_API_KEY="" # Modify this if you use wandb6 7accelerate launch \8    --num_machines $NUM_MACHINES \9    --num_processes $(( $NUM_MACHINES * $NUM_LOCAL_GPUS )) \10    --machine_rank $MACHINE_RANK \11    src/train_partcrafter.py \12        --pin_memory \13        --allow_tf32 \14$@15