Update README.md
Browse files
README.md
CHANGED
@@ -67,6 +67,7 @@ These are the key hyperparameters used during training:
|
|
67 |
* Learning rate: 1e-4
|
68 |
* Batch size: 32
|
69 |
* Gradient accumulation steps: 4
|
70 |
-
* Image resolution:
|
|
|
71 |
* Mixed-precision: fp16
|
72 |
|
|
|
67 |
* Learning rate: 1e-4
|
68 |
* Batch size: 32
|
69 |
* Gradient accumulation steps: 4
|
70 |
+
* Image resolution: 768
|
71 |
+
* Dataset size - 7k images
|
72 |
* Mixed-precision: fp16
|
73 |
|