| | --- |
| | {{ card_data }} |
| | --- |
| | |
| | <!-- This model card has been generated automatically according to the information the training script had access to. You |
| | should probably proofread and complete it, then remove this comment. --> |
| |
|
| | # {{ model_name | default("Diffusion Model") }} |
| | |
| | ## Model description |
| | |
| | This diffusion model is trained with the [🤗 Diffusers](https://github.com/huggingface/diffusers) library |
| | on the `{{ dataset_name }}` dataset. |
| |
|
| | ## Intended uses & limitations |
| |
|
| | #### How to use |
| |
|
| | ```python |
| | # TODO: add an example code snippet for running this diffusion pipeline |
| | ``` |
| |
|
| | #### Limitations and bias |
| |
|
| | [TODO: provide examples of latent issues and potential remediations] |
| |
|
| | ## Training data |
| |
|
| | [TODO: describe the data used to train the model] |
| |
|
| | ### Training hyperparameters |
| |
|
| | The following hyperparameters were used during training: |
| | - learning_rate: {{ learning_rate }} |
| | - train_batch_size: {{ train_batch_size }} |
| | - eval_batch_size: {{ eval_batch_size }} |
| | - gradient_accumulation_steps: {{ gradient_accumulation_steps }} |
| | - optimizer: AdamW with betas=({{ adam_beta1 }}, {{ adam_beta2 }}), weight_decay={{ adam_weight_decay }} and epsilon={{ adam_epsilon }} |
| | - lr_scheduler: {{ lr_scheduler }} |
| | - lr_warmup_steps: {{ lr_warmup_steps }} |
| | - ema_inv_gamma: {{ ema_inv_gamma }} |
| | - ema_inv_gamma: {{ ema_power }} |
| | - ema_inv_gamma: {{ ema_max_decay }} |
| | - mixed_precision: {{ mixed_precision }} |
| | |
| | ### Training results |
| | |
| | 📈 [TensorBoard logs](https://huggingface.co/{{ repo_name }}/tensorboard?#scalars) |
| | |
| | |
| | |
