miluELK commited on
Commit
170bd5b
1 Parent(s): f9f519e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -0
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]()