theta commited on
Commit
a47fd51
1 Parent(s): 90a7420

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -35,17 +35,17 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - learning_rate: 5e-05
39
- - train_batch_size: 8
40
- - eval_batch_size: 8
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.0
45
 
46
  ### Framework versions
47
 
48
- - Transformers 4.19.1
49
- - Pytorch 1.11.0+cu113
50
- - Datasets 2.2.1
51
  - Tokenizers 0.12.1
 
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
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
44
+ - num_epochs: 5
45
 
46
  ### Framework versions
47
 
48
+ - Transformers 4.21.1
49
+ - Pytorch 1.12.0+cu113
50
+ - Datasets 2.4.0
51
  - Tokenizers 0.12.1