From 32dd8d3f3f9f61913c2ca55de28f6f2014bf8eaa Mon Sep 17 00:00:00 2001 From: Omar Sanseviero Date: Sun, 17 Dec 2023 13:28:47 +0100 Subject: [PATCH] Nit fix to training params --- 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 97b644883cae..a124d5f32e1f 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" \