End of training

README.md

@@ -14,13 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [Ransaka/sinhala-ocr-model](https://huggingface.co/Ransaka/sinhala-ocr-model) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 5.0265
-- eval_cer: 0.4829
-- eval_runtime: 219.5849
-- eval_samples_per_second: 1.858
-- eval_steps_per_second: 0.465
-- epoch: 3.92
-- step: 300
+- Loss: 6.2306
+- Cer: 0.5161
 
 ## Model description
 
@@ -39,17 +34,35 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 3e-05
+- learning_rate: 8e-05
 - train_batch_size: 4
 - eval_batch_size: 4
 - seed: 42
-- gradient_accumulation_steps: 4
-- total_train_batch_size: 16
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 8
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- training_steps: 5000
+- training_steps: 6000
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Cer    |
+|:-------------:|:-----:|:----:|:---------------:|:------:|
+| 4.543         | 3.27  | 500  | 6.2682          | 0.7086 |
+| 2.6146        | 6.54  | 1000 | 5.8348          | 0.6390 |
+| 1.8448        | 9.8   | 1500 | 5.8076          | 0.6166 |
+| 1.3887        | 13.07 | 2000 | 6.0250          | 0.6072 |
+| 1.0271        | 16.34 | 2500 | 5.9971          | 0.5707 |
+| 0.8891        | 19.61 | 3000 | 5.9803          | 0.5630 |
+| 0.6548        | 22.88 | 3500 | 6.0045          | 0.5542 |
+| 0.4939        | 26.14 | 4000 | 6.0223          | 0.5354 |
+| 0.322         | 29.41 | 4500 | 6.1360          | 0.5233 |
+| 0.2459        | 32.68 | 5000 | 6.1166          | 0.5220 |
+| 0.123         | 35.95 | 5500 | 6.1740          | 0.5162 |
+| 0.1575        | 39.22 | 6000 | 6.2306          | 0.5161 |
+
+
 ### Framework versions
 
 - Transformers 4.35.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2133d5acdea916a9fb46a0fe510577da1dc83d74e313568753a36847fe20ef96
+oid sha256:05c49eeec5efb13e113fe9cfaf69d9afed4e9ea06a308e5296ae5f63e0f006f0
 size 1260933520

runs/Jan04_18-53-53_659516cc215a/events.out.tfevents.1704394438.659516cc215a.26.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:16ae246a8211f6d86b05b59f804dfeecfbdc346a70ce429629b315e0487078bc
-size 49969
+oid sha256:5881bfa172994f38dd639d4ce515df4aa55d0130d272d7e2ced6b88a56c8bdf0
+size 50323