aiqwe commited on
Commit
8f60fc5
1 Parent(s): 251d138

Model save

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -36,15 +36,15 @@ More information needed
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0001
39
- - train_batch_size: 8
40
  - eval_batch_size: 8
41
  - seed: 42
42
  - gradient_accumulation_steps: 2
43
- - total_train_batch_size: 16
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: cosine
46
  - lr_scheduler_warmup_ratio: 0.05
47
- - num_epochs: 10
48
 
49
  ### Training results
50
 
@@ -53,7 +53,7 @@ The following hyperparameters were used during training:
53
  ### Framework versions
54
 
55
  - PEFT 0.10.0
56
- - Transformers 4.40.1
57
  - Pytorch 2.2.1+cu121
58
  - Datasets 2.19.1
59
  - Tokenizers 0.19.1
 
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0001
39
+ - train_batch_size: 4
40
  - eval_batch_size: 8
41
  - seed: 42
42
  - gradient_accumulation_steps: 2
43
+ - total_train_batch_size: 8
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: cosine
46
  - lr_scheduler_warmup_ratio: 0.05
47
+ - num_epochs: 5
48
 
49
  ### Training results
50
 
 
53
  ### Framework versions
54
 
55
  - PEFT 0.10.0
56
+ - Transformers 4.40.2
57
  - Pytorch 2.2.1+cu121
58
  - Datasets 2.19.1
59
  - Tokenizers 0.19.1