".github/vscode:/vscode.git/clone" did not exist on "b7df4a5387fd65edf4f16fd5fc8c87a7c815a4c7"
Unverified Commit 74558ff6 authored by Omar Sanseviero's avatar Omar Sanseviero Committed by GitHub
Browse files

Nit fix to training params (#6200)

parent 49644bab
......@@ -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" \
......
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