vapogore commited on
Commit
26188b4
1 Parent(s): 2bd0522

Training complete

Browse files
Files changed (3) hide show
  1. README.md +9 -9
  2. generation_config.json +1 -1
  3. model.safetensors +1 -1
README.md CHANGED
@@ -16,11 +16,11 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # mbart-neutralization
18
 
19
- This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on the Spanish Gender Neutralization dataset (hackathon-pln-es/neutral-es ).
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0181
22
- - Bleu: 98.7341
23
- - Gen Len: 18.4896
24
 
25
  ## Model description
26
 
@@ -51,13 +51,13 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
53
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
54
- | No log | 1.0 | 440 | 0.0307 | 91.2911 | 18.25 |
55
- | 0.2343 | 2.0 | 880 | 0.0181 | 98.7341 | 18.4896 |
56
 
57
 
58
  ### Framework versions
59
 
60
- - Transformers 4.37.2
61
  - Pytorch 2.1.0+cu121
62
- - Datasets 2.17.0
63
- - Tokenizers 0.15.1
 
16
 
17
  # mbart-neutralization
18
 
19
+ This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0220
22
+ - Bleu: 98.2132
23
+ - Gen Len: 18.5417
24
 
25
  ## Model description
26
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
53
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
54
+ | No log | 1.0 | 440 | 0.0490 | 96.2659 | 19.0104 |
55
+ | 0.2462 | 2.0 | 880 | 0.0220 | 98.2132 | 18.5417 |
56
 
57
 
58
  ### Framework versions
59
 
60
+ - Transformers 4.38.1
61
  - Pytorch 2.1.0+cu121
62
+ - Datasets 2.17.1
63
+ - Tokenizers 0.15.2
generation_config.json CHANGED
@@ -7,5 +7,5 @@
7
  "max_length": 200,
8
  "num_beams": 5,
9
  "pad_token_id": 1,
10
- "transformers_version": "4.37.2"
11
  }
 
7
  "max_length": 200,
8
  "num_beams": 5,
9
  "pad_token_id": 1,
10
+ "transformers_version": "4.38.1"
11
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2b259407eeb6f8205366f376cee4602ec3421c422d51e17fe1a22fae4385ea61
3
  size 2444578688
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3fcf37a910b0311b01f94b85ab2cb38ee289434f30d446d17932891b26406438
3
  size 2444578688