Skip to content

GitLab

Unverified Commit 74558ff6 authored by Omar Sanseviero's avatar Omar Sanseviero Committed by GitHub
Browse files

Nit fix to training params (#6200)

parent 49644bab
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/en/training/unconditional_training.md
View file @ 74558ff6
...@@ -186,7 +186,7 @@ accelerate launch train_unconditional.py \ ...@@ -186,7 +186,7 @@ accelerate launch train_unconditional.py \
If you're training with more than one GPU, add the `--multi_gpu` parameter to the training command: If you're training with more than one GPU, add the `--multi_gpu` parameter to the training command:
```bash ```bash
accelerate launch --mixed_precision="fp16" --multi_gpu train_unconditional.py \ accelerate launch --multi_gpu train_unconditional.py \
--dataset_name="huggan/flowers-102-categories" \ --dataset_name="huggan/flowers-102-categories" \
--output_dir="ddpm-ema-flowers-64" \ --output_dir="ddpm-ema-flowers-64" \
--mixed_precision="fp16" \ --mixed_precision="fp16" \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment