Shresthadev403 commited on
Commit
a6ae4b3
1 Parent(s): e328432

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -32,15 +32,15 @@ More information needed
32
 
33
  The following hyperparameters were used during training:
34
  - learning_rate: 0.0005
35
- - train_batch_size: 1
36
- - eval_batch_size: 1
37
  - seed: 42
38
  - gradient_accumulation_steps: 8
39
- - total_train_batch_size: 8
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
  - lr_scheduler_type: cosine
42
  - lr_scheduler_warmup_steps: 1000
43
- - num_epochs: 1
44
 
45
  ### Training results
46
 
@@ -48,7 +48,7 @@ The following hyperparameters were used during training:
48
 
49
  ### Framework versions
50
 
51
- - Transformers 4.30.2
52
  - Pytorch 2.0.1+cu118
53
  - Datasets 2.13.1
54
  - Tokenizers 0.13.3
 
32
 
33
  The following hyperparameters were used during training:
34
  - learning_rate: 0.0005
35
+ - train_batch_size: 32
36
+ - eval_batch_size: 32
37
  - seed: 42
38
  - gradient_accumulation_steps: 8
39
+ - total_train_batch_size: 256
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
  - lr_scheduler_type: cosine
42
  - lr_scheduler_warmup_steps: 1000
43
+ - num_epochs: 10
44
 
45
  ### Training results
46
 
 
48
 
49
  ### Framework versions
50
 
51
+ - Transformers 4.31.0.dev0
52
  - Pytorch 2.0.1+cu118
53
  - Datasets 2.13.1
54
  - Tokenizers 0.13.3