anujsahani01 commited on
Commit
5594e04
1 Parent(s): 6dc9c54

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -7
README.md CHANGED
@@ -3,14 +3,14 @@ license: mit
3
  tags:
4
  - generated_from_trainer
5
  model-index:
6
- - name: finetuned_mbart
7
  results: []
8
  ---
9
 
10
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
  should probably proofread and complete it, then remove this comment. -->
12
 
13
- # finetuned_mbart
14
 
15
  This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on the None dataset.
16
 
@@ -31,17 +31,22 @@ More information needed
31
  ### Training hyperparameters
32
 
33
  The following hyperparameters were used during training:
34
- - learning_rate: 0.005
35
- - train_batch_size: 16
36
- - eval_batch_size: 16
37
  - seed: 42
38
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
39
  - lr_scheduler_type: linear
40
- - num_epochs: 1
 
 
 
 
 
41
 
42
  ### Framework versions
43
 
44
  - Transformers 4.30.2
45
  - Pytorch 2.0.1+cu118
46
- - Datasets 2.13.0
47
  - Tokenizers 0.13.3
 
3
  tags:
4
  - generated_from_trainer
5
  model-index:
6
+ - name: finetuned_Mbart
7
  results: []
8
  ---
9
 
10
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
  should probably proofread and complete it, then remove this comment. -->
12
 
13
+ # finetuned_Mbart
14
 
15
  This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on the None dataset.
16
 
 
31
  ### Training hyperparameters
32
 
33
  The following hyperparameters were used during training:
34
+ - learning_rate: 0.001
35
+ - train_batch_size: 8
36
+ - eval_batch_size: 8
37
  - seed: 42
38
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
39
  - lr_scheduler_type: linear
40
+ - lr_scheduler_warmup_steps: 50
41
+ - training_steps: 6000
42
+
43
+ ### Training results
44
+
45
+
46
 
47
  ### Framework versions
48
 
49
  - Transformers 4.30.2
50
  - Pytorch 2.0.1+cu118
51
+ - Datasets 2.13.1
52
  - Tokenizers 0.13.3