Fix output path name

This commit is contained in:
Manuel Romero 2022-06-15 21:45:49 +02:00 committed by GitHub
parent 71c6b36254
commit 57aba1ef50
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 \