Hugging Face's logo

gpriday
/
midjourney-rank

Text Classification
Transformers
PyTorch
bert
Generated from Trainer
Inference Endpoints
Model card Files Community
gpriday commited on
Commit
c27d050
•
1 Parent(s): 44b7395

End of training

Browse files
Files changed (3) hide show
  1. README.md +14 -11
  2. pytorch_model.bin +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
- license: apache-2.0
3
- base_model: bert-base-uncased
4
  tags:
5
  - generated_from_trainer
6
  model-index:
@@ -13,10 +13,10 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  # midjourney-rank
15
 
16
- This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.0497
19
- - Mse: 0.0497
20
 
21
  ## Model description
22
 
@@ -41,17 +41,20 @@ The following hyperparameters were used during training:
41
  - seed: 42
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
44
- - num_epochs: 2
45
 
46
  ### Training results
47
 
48
  | Training Loss | Epoch | Step | Validation Loss | Mse |
49
  |:-------------:|:-----:|:----:|:---------------:|:------:|
50
- | 0.0591 | 0.34 | 100 | 0.0556 | 0.0556 |
51
- | 0.0569 | 0.67 | 200 | 0.0537 | 0.0537 |
52
- | 0.0543 | 1.01 | 300 | 0.0500 | 0.0500 |
53
- | 0.0437 | 1.34 | 400 | 0.0566 | 0.0566 |
54
- | 0.0432 | 1.68 | 500 | 0.0497 | 0.0497 |
 
 
 
55
 
56
 
57
  ### Framework versions
 
1
  ---
2
+ license: mit
3
+ base_model: microsoft/MiniLM-L12-H384-uncased
4
  tags:
5
  - generated_from_trainer
6
  model-index:
 
13
 
14
  # midjourney-rank
15
 
16
+ This model is a fine-tuned version of [microsoft/MiniLM-L12-H384-uncased](https://huggingface.co/microsoft/MiniLM-L12-H384-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.0509
19
+ - Mse: 0.0509
20
 
21
  ## Model description
22
 
 
41
  - seed: 42
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
44
+ - num_epochs: 3
45
 
46
  ### Training results
47
 
48
  | Training Loss | Epoch | Step | Validation Loss | Mse |
49
  |:-------------:|:-----:|:----:|:---------------:|:------:|
50
+ | 0.061 | 0.34 | 100 | 0.0565 | 0.0565 |
51
+ | 0.0544 | 0.67 | 200 | 0.0562 | 0.0562 |
52
+ | 0.0585 | 1.01 | 300 | 0.0509 | 0.0509 |
53
+ | 0.0508 | 1.34 | 400 | 0.0523 | 0.0523 |
54
+ | 0.0508 | 1.68 | 500 | 0.0518 | 0.0518 |
55
+ | 0.0465 | 2.01 | 600 | 0.0509 | 0.0509 |
56
+ | 0.0449 | 2.35 | 700 | 0.0526 | 0.0526 |
57
+ | 0.0398 | 2.68 | 800 | 0.0509 | 0.0509 |
58
 
59
 
60
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:af16752f7acbbf164903eca1b803ac8324c6bd1890096b7b148035d8a654e63b
3
  size 133509937
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea2a89638c2ccb42bdfa8bdbff22a6a286d75ef786f3217ea689d3e00cae98d3
3
  size 133509937
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4240bbd12c11b39b5f6966a4a637e43237d0f4de13a338d020ff9cd3ca9d10b2
3
  size 3963
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c509958c6ed2cd56367fae773cba15d0886d585a8f7885ce0201295c45aaef30
3
  size 3963