PSHuman / scripts /train_768.sh
fffiloni's picture
Migrated from GitHub
2252f3d verified
raw
history blame
308 Bytes
#!/bin/bash
export WANDB_API_KEY=$KEY # replace $KEY with your wandb key
export CUDA_VISIBLE_DEVICES=0,1,2,3
### CMD
accelerate launch --config_file node_config/gpu.yaml --num_processes 4 \
train_mvdiffusion_unit_unclip.py \
--config configs/train-768-6view-onlyscan_face.yaml > log/log_$$.txt 2>&1