cs7150 / diffusers /utils /model_card_template.md
lakshyana
updated diffusers
7c8c2c8
|
raw
history blame
1.48 kB
metadata
'[object Object]': null

{{ model_name | default("Diffusion Model") }}

Model description

This diffusion model is trained with the πŸ€— Diffusers library on the {{ dataset_name }} dataset.

Intended uses & limitations

How to use

# 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)