CoolFace
Apppublic

declare-lab/tango2

sourceHugging Faceupdated 2y agoView on Hugging Face
92likes
model_card_template.md51 linesDownload Raw Back to utils
1---2{{ card_data }}3---4 5<!-- This model card has been generated automatically according to the information the training script had access to. You6should probably proofread and complete it, then remove this comment. -->7 8# {{ model_name | default("Diffusion Model") }}9 10## Model description11 12This diffusion model is trained with the [๐Ÿค— Diffusers](https://github.com/huggingface/diffusers) library 13on the `{{ dataset_name }}` dataset.14 15## Intended uses & limitations16 17#### How to use18 19```python20# TODO: add an example code snippet for running this diffusion pipeline21```22 23#### Limitations and bias24 25[TODO: provide examples of latent issues and potential remediations]26 27## Training data28 29[TODO: describe the data used to train the model]30 31### Training hyperparameters32 33The following hyperparameters were used during training:34- learning_rate: {{ learning_rate }}35- train_batch_size: {{ train_batch_size }}36- eval_batch_size: {{ eval_batch_size }}37- gradient_accumulation_steps: {{ gradient_accumulation_steps }}38- optimizer: AdamW with betas=({{ adam_beta1 }}, {{ adam_beta2 }}), weight_decay={{ adam_weight_decay }} and epsilon={{ adam_epsilon }}39- lr_scheduler: {{ lr_scheduler }}40- lr_warmup_steps: {{ lr_warmup_steps }}41- ema_inv_gamma: {{ ema_inv_gamma }}42- ema_inv_gamma: {{ ema_power }}43- ema_inv_gamma: {{ ema_max_decay }}44- mixed_precision: {{ mixed_precision }}45 46### Training results47 48๐Ÿ“ˆ [TensorBoard logs](https://huggingface.co/{{ repo_name }}/tensorboard?#scalars)49 50 51