juansgultom commited on
Commit
ca3c031
1 Parent(s): eecd784

End of training

Browse files
Files changed (2) hide show
  1. README.md +4 -4
  2. generation_config.json +1 -1
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [microsoft/git-base](https://huggingface.co/microsoft/git-base) on the imagefolder dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 6.6548
21
- - Wer Score: 8.8
22
 
23
  ## Model description
24
 
@@ -52,12 +52,12 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Wer Score |
54
  |:-------------:|:-----:|:----:|:---------------:|:---------:|
55
- | 3.9097 | 50.0 | 50 | 6.6548 | 8.8 |
56
 
57
 
58
  ### Framework versions
59
 
60
- - Transformers 4.41.1
61
  - Pytorch 2.3.0+cu121
62
  - Datasets 2.19.2
63
  - Tokenizers 0.19.1
 
17
 
18
  This model is a fine-tuned version of [microsoft/git-base](https://huggingface.co/microsoft/git-base) on the imagefolder dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 6.7529
21
+ - Wer Score: 8.92
22
 
23
  ## Model description
24
 
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Wer Score |
54
  |:-------------:|:-----:|:----:|:---------------:|:---------:|
55
+ | 3.844 | 50.0 | 50 | 6.7529 | 8.92 |
56
 
57
 
58
  ### Framework versions
59
 
60
+ - Transformers 4.41.2
61
  - Pytorch 2.3.0+cu121
62
  - Datasets 2.19.2
63
  - Tokenizers 0.19.1
generation_config.json CHANGED
@@ -3,5 +3,5 @@
3
  "bos_token_id": 101,
4
  "eos_token_id": 102,
5
  "pad_token_id": 0,
6
- "transformers_version": "4.41.1"
7
  }
 
3
  "bos_token_id": 101,
4
  "eos_token_id": 102,
5
  "pad_token_id": 0,
6
+ "transformers_version": "4.41.2"
7
  }