File size: 619 Bytes
170bd5b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
---
license: afl-3.0
datasets:
- huggan/few-shot-pokemon
language:
- en
library_name: diffusers
---
## Training data
huggan/few-shot-pokemon
### Training hyperparameters
The following hyperparameters were used during training:
--checkpointing_steps=1000 \
--dataset_name="huggan/few-shot-pokemon" \
--resolution=128 \
--output_dir="ddpm-ema-pokemon-128" \
--train_batch_size=16 \
--eval_batch_size=16 \
--num_epochs=100 \
--gradient_accumulation_steps=1 \
--learning_rate=1e-4 \
--lr_warmup_steps=800 \
--mixed_precision="fp16" \
--push_to_hub
### Training results
📈 [TensorBoard logs]() |