From 74558ff65b0ed169cc5f36b05d95b4cb320e907c Mon Sep 17 00:00:00 2001 From: Omar Sanseviero Date: Mon, 18 Dec 2023 11:06:16 +0100 Subject: [PATCH] Nit fix to training params (#6200) --- docs/source/en/training/unconditional_training.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/training/unconditional_training.md b/docs/source/en/training/unconditional_training.md index 97b644883c..a124d5f32e 100644 --- a/docs/source/en/training/unconditional_training.md +++ b/docs/source/en/training/unconditional_training.md @@ -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: ```bash -accelerate launch --mixed_precision="fp16" --multi_gpu train_unconditional.py \ +accelerate launch --multi_gpu train_unconditional.py \ --dataset_name="huggan/flowers-102-categories" \ --output_dir="ddpm-ema-flowers-64" \ --mixed_precision="fp16" \