Hugging Face's logo

cotysong113
/
my_awesome_eli5_mlm_model

Fill-Mask
Transformers
TensorBoard
Safetensors
roberta
Generated from Trainer
Model card Files
xet
 Metrics Community
cotysong113 commited on
Commit
f14c928
·
verified ·
1 Parent(s): 21f4397

End of training

Browse files
Files changed (3) hide show
  1. README.md +6 -6
  2. model.safetensors +1 -1
  3. training_args.bin +2 -2
README.md CHANGED
@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [distilbert/distilroberta-base](https://huggingface.co/distilbert/distilroberta-base) on the eli5_category dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 2.0244
22
 
23
  ## Model description
24
 
@@ -49,14 +49,14 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
- | 2.2961 | 1.0 | 946 | 2.0529 |
53
- | 2.1976 | 2.0 | 1892 | 2.0641 |
54
- | 2.1467 | 3.0 | 2838 | 1.9921 |
55
 
56
 
57
  ### Framework versions
58
 
59
  - Transformers 4.47.0
60
- - Pytorch 2.5.1
61
- - Datasets 3.1.0
62
  - Tokenizers 0.21.0
 
18
 
19
  This model is a fine-tuned version of [distilbert/distilroberta-base](https://huggingface.co/distilbert/distilroberta-base) on the eli5_category dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 2.0082
22
 
23
  ## Model description
24
 
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
+ | 2.3019 | 1.0 | 946 | 2.0683 |
53
+ | 2.1768 | 2.0 | 1892 | 2.0422 |
54
+ | 2.1308 | 3.0 | 2838 | 2.0112 |
55
 
56
 
57
  ### Framework versions
58
 
59
  - Transformers 4.47.0
60
+ - Pytorch 2.5.1+cu124
61
+ - Datasets 2.21.0
62
  - Tokenizers 0.21.0
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3a663e46aa932bb65ba9ba2724276170d74dc0c797d6d42f389ad3b0adc28e44
3
  size 328693404
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5da4f4e772fc92ef570c8bf03053ba8e1c831a776389b73c3f93e33a44ba44ff
3
  size 328693404
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9ed40c7a6bf987cf181a363fd87469d48ae930354a0b3e9e3f088cba6c65fd97
3
- size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:384ba85f8ff3a7cc58ff812b63a23d9f448f6d7f6d8745212c1127a9a7791d97
3
+ size 5304