Hugging Face's logo

bmd1905
/
mamba_text_classification

Transformers
PyTorch
Generated from Trainer
Inference Endpoints
Model card Files Community
bmd1905 commited on
Commit
ad43750
•
1 Parent(s): 6b12dd5

Training complete

Browse files
Files changed (2) hide show
  1. README.md +15 -15
  2. pytorch_model.bin +1 -1
README.md CHANGED
@@ -15,8 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model was trained from scratch on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.2112
19
- - Accuracy: 0.9454
20
 
21
  ## Model description
22
 
@@ -48,21 +48,21 @@ The following hyperparameters were used during training:
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
- | 0.0067 | 0.1 | 625 | 0.3108 | 0.9158 |
52
- | 1.8154 | 0.2 | 1250 | 0.2347 | 0.931 |
53
- | 0.0341 | 0.3 | 1875 | 0.2995 | 0.9246 |
54
- | 0.0052 | 0.4 | 2500 | 0.2312 | 0.9364 |
55
- | 0.0001 | 0.5 | 3125 | 0.2634 | 0.932 |
56
- | 0.0016 | 0.6 | 3750 | 0.2250 | 0.942 |
57
- | 1.8912 | 0.7 | 4375 | 0.2112 | 0.9452 |
58
- | 0.0025 | 0.8 | 5000 | 0.2191 | 0.946 |
59
- | 0.0149 | 0.9 | 5625 | 0.2125 | 0.945 |
60
- | 0.0035 | 1.0 | 6250 | 0.2112 | 0.9454 |
61
 
62
 
63
  ### Framework versions
64
 
65
- - Transformers 4.35.2
66
- - Pytorch 2.1.0+cu121
67
- - Datasets 2.17.0
68
  - Tokenizers 0.15.2
 
15
 
16
  This model was trained from scratch on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.2351
19
+ - Accuracy: 0.9406
20
 
21
  ## Model description
22
 
 
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | 0.0061 | 0.1 | 625 | 0.2807 | 0.9074 |
52
+ | 1.7561 | 0.2 | 1250 | 0.2201 | 0.9252 |
53
+ | 0.0296 | 0.3 | 1875 | 0.3153 | 0.9184 |
54
+ | 0.0057 | 0.4 | 2500 | 0.2213 | 0.9342 |
55
+ | 0.0002 | 0.5 | 3125 | 0.2311 | 0.935 |
56
+ | 2.8004 | 0.6 | 3750 | 0.2264 | 0.9378 |
57
+ | 0.0143 | 0.7 | 4375 | 0.2599 | 0.9348 |
58
+ | 0.0021 | 0.8 | 5000 | 0.2267 | 0.939 |
59
+ | 0.2814 | 0.9 | 5625 | 0.2343 | 0.942 |
60
+ | 0.069 | 1.0 | 6250 | 0.2351 | 0.9406 |
61
 
62
 
63
  ### Framework versions
64
 
65
+ - Transformers 4.38.1
66
+ - Pytorch 2.1.1+cu118
67
+ - Datasets 2.17.1
68
  - Tokenizers 0.15.2
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9bfc282efc03ee887705f20bdb69adb36074408c1fcf7163c3ab170e8c099a67
3
  size 516640282
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e99ae76254ee29a94990c1c941e1bc9c01874b10e33715d1b6aced571489e23
3
  size 516640282