jschmock commited on
Commit
2fe674c
1 Parent(s): 3b99551

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -14
README.md CHANGED
@@ -15,8 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [svalabs/gbert-large-zeroshot-nli](https://huggingface.co/svalabs/gbert-large-zeroshot-nli) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.0022
19
- - F1: 1.0
20
 
21
  ## Model description
22
 
@@ -36,11 +36,11 @@ More information needed
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 5e-05
39
- - train_batch_size: 16
40
  - eval_batch_size: 8
41
  - seed: 42
42
  - gradient_accumulation_steps: 4
43
- - total_train_batch_size: 64
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - num_epochs: 10
@@ -49,16 +49,16 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | F1 |
51
  |:-------------:|:-----:|:----:|:---------------:|:------:|
52
- | No log | 0.94 | 11 | 0.6783 | 0.7786 |
53
- | No log | 1.94 | 22 | 0.1718 | 0.9272 |
54
- | No log | 2.94 | 33 | 0.0769 | 0.9887 |
55
- | No log | 3.94 | 44 | 0.0686 | 0.9887 |
56
- | No log | 4.94 | 55 | 0.0227 | 0.9887 |
57
- | No log | 5.94 | 66 | 0.0075 | 1.0 |
58
- | No log | 6.94 | 77 | 0.0099 | 1.0 |
59
- | No log | 7.94 | 88 | 0.0025 | 1.0 |
60
- | No log | 8.94 | 99 | 0.0023 | 1.0 |
61
- | No log | 9.94 | 110 | 0.0022 | 1.0 |
62
 
63
 
64
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [svalabs/gbert-large-zeroshot-nli](https://huggingface.co/svalabs/gbert-large-zeroshot-nli) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.1126
19
+ - F1: 0.9887
20
 
21
  ## Model description
22
 
 
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 5e-05
39
+ - train_batch_size: 4
40
  - eval_batch_size: 8
41
  - seed: 42
42
  - gradient_accumulation_steps: 4
43
+ - total_train_batch_size: 16
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - num_epochs: 10
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | F1 |
51
  |:-------------:|:-----:|:----:|:---------------:|:------:|
52
+ | No log | 0.98 | 46 | 0.3264 | 0.9074 |
53
+ | No log | 1.98 | 92 | 0.1266 | 0.9590 |
54
+ | No log | 2.98 | 138 | 0.0603 | 0.9887 |
55
+ | No log | 3.98 | 184 | 0.1000 | 0.9887 |
56
+ | No log | 4.98 | 230 | 0.1075 | 0.9887 |
57
+ | No log | 5.98 | 276 | 0.1091 | 0.9887 |
58
+ | No log | 6.98 | 322 | 0.1109 | 0.9887 |
59
+ | No log | 7.98 | 368 | 0.1119 | 0.9887 |
60
+ | No log | 8.98 | 414 | 0.1124 | 0.9887 |
61
+ | No log | 9.98 | 460 | 0.1126 | 0.9887 |
62
 
63
 
64
  ### Framework versions