Kunalmod commited on
Commit
9858e20
verified
1 Parent(s): 065fbb1

Model save

Browse files
Files changed (1) hide show
  1. README.md +6 -5
README.md CHANGED
@@ -1,4 +1,5 @@
1
  ---
 
2
  license: cc-by-4.0
3
  base_model: deepset/roberta-base-squad2
4
  tags:
@@ -34,11 +35,11 @@ More information needed
34
  The following hyperparameters were used during training:
35
  - learning_rate: 2e-05
36
  - train_batch_size: 8
37
- - eval_batch_size: 8
38
  - seed: 42
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
40
  - lr_scheduler_type: linear
41
- - num_epochs: 5
42
 
43
  ### Training results
44
 
@@ -46,6 +47,6 @@ The following hyperparameters were used during training:
46
 
47
  ### Framework versions
48
 
49
- - Transformers 4.38.2
50
- - Pytorch 2.2.1+cu121
51
- - Tokenizers 0.15.2
 
1
  ---
2
+ library_name: transformers
3
  license: cc-by-4.0
4
  base_model: deepset/roberta-base-squad2
5
  tags:
 
35
  The following hyperparameters were used during training:
36
  - learning_rate: 2e-05
37
  - train_batch_size: 8
38
+ - eval_batch_size: 32
39
  - seed: 42
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
  - lr_scheduler_type: linear
42
+ - num_epochs: 8
43
 
44
  ### Training results
45
 
 
47
 
48
  ### Framework versions
49
 
50
+ - Transformers 4.44.2
51
+ - Pytorch 2.4.0+cu121
52
+ - Tokenizers 0.19.1