End of training

README.md

@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [microsoft/git-base](https://huggingface.co/microsoft/git-base) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0621
-- Wer Score: 4.7174
+- Loss: 0.0444
+- Wer Score: 10.0470
 
 ## Model description
 
@@ -51,24 +51,30 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Wer Score |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|
-| No log        | 1.48  | 20   | 7.3896          | 132.1522  |
-| No log        | 2.96  | 40   | 5.5234          | 1.1413    |
-| No log        | 4.44  | 60   | 3.7317          | 1.1196    |
-| No log        | 5.93  | 80   | 2.1477          | 1.1304    |
-| No log        | 7.41  | 100  | 0.9848          | 1.1087    |
-| No log        | 8.89  | 120  | 0.3860          | 1.1304    |
-| No log        | 10.37 | 140  | 0.1720          | 1.1087    |
-| No log        | 11.85 | 160  | 0.1022          | 1.1196    |
-| No log        | 13.33 | 180  | 0.0745          | 1.0543    |
-| No log        | 14.81 | 200  | 0.0665          | 2.9239    |
-| No log        | 16.3  | 220  | 0.0631          | 2.7174    |
-| No log        | 17.78 | 240  | 0.0622          | 2.5326    |
-| No log        | 19.26 | 260  | 0.0621          | 4.7174    |
+| No log        | 1.05  | 20   | 7.3069          | 113.3758  |
+| No log        | 2.11  | 40   | 5.3566          | 3.2282    |
+| No log        | 3.16  | 60   | 3.4409          | 1.1879    |
+| No log        | 4.21  | 80   | 1.7218          | 1.1007    |
+| No log        | 5.26  | 100  | 0.5834          | 1.0872    |
+| No log        | 6.32  | 120  | 0.1684          | 1.3020    |
+| No log        | 7.37  | 140  | 0.0720          | 2.9732    |
+| No log        | 8.42  | 160  | 0.0507          | 2.0805    |
+| No log        | 9.47  | 180  | 0.0467          | 3.0336    |
+| No log        | 10.53 | 200  | 0.0415          | 10.6107   |
+| No log        | 11.58 | 220  | 0.0425          | 7.7383    |
+| No log        | 12.63 | 240  | 0.0426          | 14.1745   |
+| No log        | 13.68 | 260  | 0.0434          | 6.0067    |
+| No log        | 14.74 | 280  | 0.0447          | 10.5503   |
+| No log        | 15.79 | 300  | 0.0434          | 9.1678    |
+| No log        | 16.84 | 320  | 0.0439          | 10.8591   |
+| No log        | 17.89 | 340  | 0.0446          | 10.0470   |
+| No log        | 18.95 | 360  | 0.0444          | 10.1208   |
+| No log        | 20.0  | 380  | 0.0444          | 10.0470   |
 
 
 ### Framework versions
 
-- Transformers 4.34.0
-- Pytorch 2.0.1+cu118
+- Transformers 4.34.1
+- Pytorch 2.1.0+cu118
 - Datasets 2.14.5
 - Tokenizers 0.14.1

generation_config.json

@@ -3,5 +3,5 @@
   "bos_token_id": 101,
   "eos_token_id": 102,
   "pad_token_id": 0,
-  "transformers_version": "4.34.0"
+  "transformers_version": "4.34.1"
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30381d9ca6b7bd815246b6709706de9144bcfd1b886375968bbb4a5731365e59
+oid sha256:3138cb8183406e38883623021f6c6e35f6808f2b3313636e920d11a984bda84d
 size 706584718