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