| 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | |
| 4 | # NCCL_NVLS_ENABLE=0 accelerate launch train_scalp_diffusion.py --config ./configs/config_scalp_texture_conditional.json --batch-size 4 --mixed-precision bf16 --use-tensorboard --save-checkpoints --compile --save-every 100000 --name scalp_diffusion --dataset_path=<PATH_DIFFLOCKS> --dataset_processed_path=<PATH_DIFFLOCKS_PROCESSED> |