abhikr487 commited on
Commit
3f241eb
·
verified ·
1 Parent(s): 79d449b

https://huggingface.co/abhikr487/lab1_finetune

Browse files
Files changed (2) hide show
  1. README.md +5 -5
  2. generation_config.json +1 -1
README.md CHANGED
@@ -22,7 +22,7 @@ model-index:
22
  metrics:
23
  - name: Bleu
24
  type: bleu
25
- value: 52.930569776237235
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [Helsinki-NLP/opus-mt-en-fr](https://huggingface.co/Helsinki-NLP/opus-mt-en-fr) on the kde4 dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.8552
36
- - Bleu: 52.9306
37
 
38
  ## Model description
39
 
@@ -52,7 +52,7 @@ More information needed
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
- - learning_rate: 2e-05
56
  - train_batch_size: 32
57
  - eval_batch_size: 64
58
  - seed: 42
@@ -67,7 +67,7 @@ The following hyperparameters were used during training:
67
 
68
  ### Framework versions
69
 
70
- - Transformers 4.37.2
71
  - Pytorch 2.1.0+cu121
72
  - Datasets 2.17.0
73
  - Tokenizers 0.15.2
 
22
  metrics:
23
  - name: Bleu
24
  type: bleu
25
+ value: 54.58730387905399
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
32
 
33
  This model is a fine-tuned version of [Helsinki-NLP/opus-mt-en-fr](https://huggingface.co/Helsinki-NLP/opus-mt-en-fr) on the kde4 dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.8057
36
+ - Bleu: 54.5873
37
 
38
  ## Model description
39
 
 
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
+ - learning_rate: 5e-05
56
  - train_batch_size: 32
57
  - eval_batch_size: 64
58
  - seed: 42
 
67
 
68
  ### Framework versions
69
 
70
+ - Transformers 4.35.2
71
  - Pytorch 2.1.0+cu121
72
  - Datasets 2.17.0
73
  - Tokenizers 0.15.2
generation_config.json CHANGED
@@ -12,5 +12,5 @@
12
  "num_beams": 4,
13
  "pad_token_id": 59513,
14
  "renormalize_logits": true,
15
- "transformers_version": "4.37.2"
16
  }
 
12
  "num_beams": 4,
13
  "pad_token_id": 59513,
14
  "renormalize_logits": true,
15
+ "transformers_version": "4.35.2"
16
  }