Go2Heart commited on
Commit
a30537a
1 Parent(s): 37c2de4

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -11
README.md CHANGED
@@ -16,7 +16,12 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the squad_v2 dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 1.4311
 
 
 
 
 
20
 
21
  ## Model description
22
 
@@ -35,7 +40,7 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - learning_rate: 2e-05
39
  - train_batch_size: 16
40
  - eval_batch_size: 16
41
  - seed: 42
@@ -43,15 +48,6 @@ The following hyperparameters were used during training:
43
  - lr_scheduler_type: linear
44
  - num_epochs: 3
45
 
46
- ### Training results
47
-
48
- | Training Loss | Epoch | Step | Validation Loss |
49
- |:-------------:|:-----:|:-----:|:---------------:|
50
- | 1.8729 | 1.0 | 8235 | 1.6423 |
51
- | 1.6782 | 2.0 | 16470 | 1.4738 |
52
- | 1.6075 | 3.0 | 24705 | 1.4311 |
53
-
54
-
55
  ### Framework versions
56
 
57
  - Transformers 4.20.1
 
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the squad_v2 dataset.
18
  It achieves the following results on the evaluation set:
19
+ - eval_loss: 1.1605
20
+ - eval_runtime: 71.6663
21
+ - eval_samples_per_second: 169.312
22
+ - eval_steps_per_second: 10.591
23
+ - epoch: 2.0
24
+ - step: 16470
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - learning_rate: 0.002
44
  - train_batch_size: 16
45
  - eval_batch_size: 16
46
  - seed: 42
 
48
  - lr_scheduler_type: linear
49
  - num_epochs: 3
50
 
 
 
 
 
 
 
 
 
 
51
  ### Framework versions
52
 
53
  - Transformers 4.20.1