Update README.md
Browse files
README.md
CHANGED
|
@@ -37,9 +37,9 @@ image.save("my_image.png")
|
|
| 37 |
|
| 38 |
These are the key hyperparameters used during training:
|
| 39 |
|
| 40 |
-
* Epochs:
|
| 41 |
* Learning rate: 1e-06
|
| 42 |
-
* Batch size:
|
| 43 |
* Gradient accumulation steps: 4
|
| 44 |
* Image resolution: 224
|
| 45 |
* Mixed-precision: fp16
|
|
|
|
| 37 |
|
| 38 |
These are the key hyperparameters used during training:
|
| 39 |
|
| 40 |
+
* Epochs: 5
|
| 41 |
* Learning rate: 1e-06
|
| 42 |
+
* Batch size: 4
|
| 43 |
* Gradient accumulation steps: 4
|
| 44 |
* Image resolution: 224
|
| 45 |
* Mixed-precision: fp16
|