mrm8488 commited on
Commit
e3a3c25
1 Parent(s): 5f63a7e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +65 -0
README.md ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ library_name: diffusers
5
+ tags: []
6
+ datasets: huggan/smithsonian_butterflies_subset
7
+ metrics: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
12
+
13
+ # ddpm-ema-flower-64
14
+
15
+ ## Model description
16
+
17
+ This diffusion model is trained with the [🤗 Diffusers](https://github.com/huggingface/diffusers) library
18
+ on the `huggan/smithsonian_butterflies_subset` dataset.
19
+
20
+ ## Intended uses & limitations
21
+
22
+ #### How to use
23
+
24
+ ```python
25
+ from diffusers import DDPMPipeline
26
+
27
+ model_id = "mrm8488/ddpm-ema-butterflies-128"
28
+
29
+ # load model and scheduler
30
+ pipeline = DDPMPipeline.from_pretrained(model_id)
31
+
32
+ # run pipeline in inference
33
+ image = pipeline()["sample"]
34
+
35
+ # save image
36
+ image[0].save("butterfly.png")
37
+ ```
38
+
39
+ #### Limitations and bias
40
+
41
+ [TODO: provide examples of latent issues and potential remediations]
42
+
43
+ ## Training data
44
+
45
+ [TODO: describe the data used to train the model]
46
+
47
+ ### Training hyperparameters
48
+
49
+ The following hyperparameters were used during training:
50
+ - learning_rate: 0.0001
51
+ - train_batch_size: 256
52
+ - eval_batch_size: 128
53
+ - gradient_accumulation_steps: 1
54
+ - optimizer: AdamW with betas=(0.95, 0.999), weight_decay=1e-06 and epsilon=1e-08
55
+ - lr_scheduler: None
56
+ - lr_warmup_steps: 500
57
+ - ema_inv_gamma: 1.0
58
+ - ema_inv_gamma: 0.75
59
+ - ema_inv_gamma: 0.9999
60
+ - mixed_precision: fp16
61
+
62
+ ### Training results
63
+
64
+ 📈 [TensorBoard logs](https://huggingface.co/mrm8488/ddpm-ema-butterflies-128/tensorboard?#scalars)
65
+