stojchet commited on
Commit
e94a7cb
·
verified ·
1 Parent(s): f9ec597

End of training

Browse files
Files changed (2) hide show
  1. README.md +5 -5
  2. generation_config.json +1 -1
README.md CHANGED
@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [deepseek-ai/deepseek-coder-1.3b-base](https://huggingface.co/deepseek-ai/deepseek-coder-1.3b-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.4712
22
 
23
  ## Model description
24
 
@@ -43,7 +43,7 @@ The following hyperparameters were used during training:
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: 3
@@ -52,12 +52,12 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss |
54
  |:-------------:|:------:|:----:|:---------------:|
55
- | 0.5183 | 2.3232 | 100 | 0.4712 |
56
 
57
 
58
  ### Framework versions
59
 
60
- - Transformers 4.50.0.dev0
61
  - Pytorch 2.5.1+cu124
62
  - Datasets 2.19.2
63
- - Tokenizers 0.21.0
 
18
 
19
  This model is a fine-tuned version of [deepseek-ai/deepseek-coder-1.3b-base](https://huggingface.co/deepseek-ai/deepseek-coder-1.3b-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.4710
22
 
23
  ## Model description
24
 
 
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
  - lr_scheduler_warmup_steps: 200
49
  - num_epochs: 3
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss |
54
  |:-------------:|:------:|:----:|:---------------:|
55
+ | 0.509 | 2.3088 | 100 | 0.4710 |
56
 
57
 
58
  ### Framework versions
59
 
60
+ - Transformers 4.45.0
61
  - Pytorch 2.5.1+cu124
62
  - Datasets 2.19.2
63
+ - Tokenizers 0.20.3
generation_config.json CHANGED
@@ -2,5 +2,5 @@
2
  "_from_model_config": true,
3
  "bos_token_id": 32013,
4
  "eos_token_id": 32014,
5
- "transformers_version": "4.50.0.dev0"
6
  }
 
2
  "_from_model_config": true,
3
  "bos_token_id": 32013,
4
  "eos_token_id": 32014,
5
+ "transformers_version": "4.45.0"
6
  }