Merge pull request #15 from mrm8488/patch-1

Fix output path name
This commit is contained in:
Anton Lozhkov 2022-06-15 22:52:24 +02:00 committed by GitHub
commit 7d4bafa8a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

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