End of training
Browse files- README.md +3 -3
- val_imgs_grid.png +0 -0
README.md
CHANGED
@@ -37,12 +37,12 @@ image.save("my_image.png")
|
|
37 |
|
38 |
These are the key hyperparameters used during training:
|
39 |
|
40 |
-
* Epochs:
|
41 |
* Learning rate: 1e-05
|
42 |
* Batch size: 16
|
43 |
* Gradient accumulation steps: 4
|
44 |
* Image resolution: 512
|
45 |
-
* Mixed-precision:
|
46 |
|
47 |
|
48 |
-
More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/vishnou/text2image-fine-tune/runs/
|
|
|
37 |
|
38 |
These are the key hyperparameters used during training:
|
39 |
|
40 |
+
* Epochs: 1
|
41 |
* Learning rate: 1e-05
|
42 |
* Batch size: 16
|
43 |
* Gradient accumulation steps: 4
|
44 |
* Image resolution: 512
|
45 |
+
* Mixed-precision: bf16
|
46 |
|
47 |
|
48 |
+
More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/vishnou/text2image-fine-tune/runs/dnbasjwq).
|
val_imgs_grid.png
CHANGED