dreambooth_teddy / README.md
danwaldie's picture
Update README.md
19da5cb
|
raw
history blame
2.09 kB
---
library_name: keras
license: apache-2.0
datasets:
- danwaldie/dreambooth
pipeline_tag: text-to-image
tags:
- keras-dreambooth
- nature
---
## Model description
Dreambooth fine-tuned for pictures of my dog, Teddy. Prompts with "teddy_holmes dog" as the unique identifier + class label.
The model seems overtrained on the poses supplied in the training images.
### About Teddy
Teddy is a Mini Double Doodle. He's the offspring of a Labradoodle and a Mini Golden Doodle, so he is one half Poodle, one quarter Labrador Retriever, and one quarter Golden Retriever.
## Examples
![Teddy as an Astronaut](Teddy-Astronaut.png)
![Teddy as a Knight](Teddy-Knight.png)
![Teddy on a floatie](Teddy-Pineapple.png)
![Teddy as a Superhero](Teddy-Superhero.png)
## Intended uses & limitations
Anyone is welcome to use the model if they want to generate pictures of an adorable dog doing interesting things.
## Training and evaluation data
More information needed
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
| Hyperparameters | Value |
| :-- | :-- |
| inner_optimizer.class_name | Custom>RMSprop |
| inner_optimizer.config.name | RMSprop |
| inner_optimizer.config.weight_decay | None |
| inner_optimizer.config.clipnorm | None |
| inner_optimizer.config.global_clipnorm | None |
| inner_optimizer.config.clipvalue | None |
| inner_optimizer.config.use_ema | False |
| inner_optimizer.config.ema_momentum | 0.99 |
| inner_optimizer.config.ema_overwrite_frequency | 100 |
| inner_optimizer.config.jit_compile | True |
| inner_optimizer.config.is_legacy_optimizer | False |
| inner_optimizer.config.learning_rate | 0.0010000000474974513 |
| inner_optimizer.config.rho | 0.9 |
| inner_optimizer.config.momentum | 0.0 |
| inner_optimizer.config.epsilon | 1e-07 |
| inner_optimizer.config.centered | False |
| dynamic | True |
| initial_scale | 32768.0 |
| dynamic_growth_steps | 2000 |
| training_precision | mixed_float16 |
## Model Plot
<details>
<summary>View Model Plot</summary>
![Model Image](./model.png)
</details>