1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Update README.md

This commit is contained in:
Anton Lozhkov
2022-06-22 13:52:45 +02:00
committed by GitHub
parent 0d80fe9327
commit 33abc79515

View File

@@ -10,7 +10,7 @@ python -m torch.distributed.launch \
train_unconditional.py \
--dataset="huggan/flowers-102-categories" \
--resolution=64 \
--output_path="flowers-ddpm" \
--output_dir="flowers-ddpm" \
--batch_size=16 \
--num_epochs=100 \
--gradient_accumulation_steps=1 \
@@ -34,7 +34,7 @@ python -m torch.distributed.launch \
train_unconditional.py \
--dataset="huggan/pokemon" \
--resolution=64 \
--output_path="pokemon-ddpm" \
--output_dir="pokemon-ddpm" \
--batch_size=16 \
--num_epochs=100 \
--gradient_accumulation_steps=1 \