stojchet commited on
Commit
8255bfc
·
verified ·
1 Parent(s): 23d4848

End of training

Browse files
Files changed (2) hide show
  1. README.md +13 -9
  2. generation_config.json +6 -0
README.md CHANGED
@@ -1,11 +1,11 @@
1
  ---
 
2
  license: other
3
- library_name: peft
4
  tags:
5
  - trl
6
  - sft
7
  - generated_from_trainer
8
- base_model: deepseek-ai/deepseek-coder-1.3b-base
9
  datasets:
10
  - generator
11
  model-index:
@@ -37,20 +37,24 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - learning_rate: 0.00141
41
  - train_batch_size: 8
42
  - eval_batch_size: 8
43
  - seed: 42
44
  - gradient_accumulation_steps: 16
45
  - total_train_batch_size: 128
46
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
 
48
  - num_epochs: 1
49
 
 
 
 
 
50
  ### Framework versions
51
 
52
- - PEFT 0.10.0
53
- - Transformers 4.42.0.dev0
54
- - Pytorch 2.2.2+cu121
55
- - Datasets 2.19.0
56
- - Tokenizers 0.19.1
 
1
  ---
2
+ library_name: transformers
3
  license: other
4
+ base_model: deepseek-ai/deepseek-coder-1.3b-base
5
  tags:
6
  - trl
7
  - sft
8
  - generated_from_trainer
 
9
  datasets:
10
  - generator
11
  model-index:
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 0.1
41
  - train_batch_size: 8
42
  - eval_batch_size: 8
43
  - seed: 42
44
  - gradient_accumulation_steps: 16
45
  - total_train_batch_size: 128
46
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
47
  - lr_scheduler_type: linear
48
+ - lr_scheduler_warmup_steps: 200
49
  - num_epochs: 1
50
 
51
+ ### Training results
52
+
53
+
54
+
55
  ### Framework versions
56
 
57
+ - Transformers 4.50.0.dev0
58
+ - Pytorch 2.5.1+cu124
59
+ - Datasets 2.19.2
60
+ - Tokenizers 0.21.1
 
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 32013,
4
+ "eos_token_id": 32014,
5
+ "transformers_version": "4.50.0.dev0"
6
+ }