stojchet commited on
Commit
d6adb76
·
verified ·
1 Parent(s): 848d55f

End of training

Browse files
Files changed (2) hide show
  1. README.md +11 -10
  2. generation_config.json +6 -0
README.md CHANGED
@@ -1,9 +1,7 @@
1
  ---
2
- base_model: deepseek-ai/deepseek-coder-1.3b-base
3
- datasets:
4
- - generator
5
- library_name: peft
6
  license: other
 
7
  tags:
8
  - trl
9
  - sft
@@ -16,10 +14,11 @@ model-index:
16
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
  should probably proofread and complete it, then remove this comment. -->
18
 
19
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/stojchets/huggingface/runs/yc3e0509)
20
  # sft2
21
 
22
- 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 the generator dataset.
 
 
23
 
24
  ## Model description
25
 
@@ -51,12 +50,14 @@ The following hyperparameters were used during training:
51
 
52
  ### Training results
53
 
 
 
 
54
 
55
 
56
  ### Framework versions
57
 
58
- - PEFT 0.10.0
59
- - Transformers 4.43.0.dev0
60
- - Pytorch 2.2.2+cu121
61
  - Datasets 2.19.2
62
- - 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
 
14
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
  should probably proofread and complete it, then remove this comment. -->
16
 
 
17
  # sft2
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
 
 
50
 
51
  ### Training results
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 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.45.0"
6
+ }