baldoino commited on
Commit
d7ea7bd
·
verified ·
1 Parent(s): b8f9372

End of training

Browse files
Files changed (1) hide show
  1. README.md +9 -9
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
- base_model: bigcode/starcoderbase-1b
3
  library_name: peft
4
  license: bigcode-openrail-m
 
5
  tags:
6
  - generated_from_trainer
7
  model-index:
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [bigcode/starcoderbase-1b](https://huggingface.co/bigcode/starcoderbase-1b) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 1.1005
20
 
21
  ## Model description
22
 
@@ -36,10 +36,10 @@ More information needed
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0005
39
- - train_batch_size: 16
40
- - eval_batch_size: 16
41
  - seed: 42
42
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: cosine
44
  - lr_scheduler_warmup_steps: 30
45
  - training_steps: 100
@@ -48,13 +48,13 @@ The following hyperparameters were used during training:
48
 
49
  | Training Loss | Epoch | Step | Validation Loss |
50
  |:-------------:|:-----:|:----:|:---------------:|
51
- | 1.2344 | 1.0 | 100 | 1.1005 |
52
 
53
 
54
  ### Framework versions
55
 
56
  - PEFT 0.13.2
57
- - Transformers 4.44.2
58
- - Pytorch 2.5.0+cu121
59
  - Datasets 3.1.0
60
- - Tokenizers 0.19.1
 
1
  ---
 
2
  library_name: peft
3
  license: bigcode-openrail-m
4
+ base_model: bigcode/starcoderbase-1b
5
  tags:
6
  - generated_from_trainer
7
  model-index:
 
16
 
17
  This model is a fine-tuned version of [bigcode/starcoderbase-1b](https://huggingface.co/bigcode/starcoderbase-1b) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.9177
20
 
21
  ## Model description
22
 
 
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0005
39
+ - train_batch_size: 4
40
+ - eval_batch_size: 4
41
  - seed: 42
42
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
43
  - lr_scheduler_type: cosine
44
  - lr_scheduler_warmup_steps: 30
45
  - training_steps: 100
 
48
 
49
  | Training Loss | Epoch | Step | Validation Loss |
50
  |:-------------:|:-----:|:----:|:---------------:|
51
+ | 1.0155 | 1.0 | 100 | 0.9177 |
52
 
53
 
54
  ### Framework versions
55
 
56
  - PEFT 0.13.2
57
+ - Transformers 4.46.2
58
+ - Pytorch 2.5.1+cu124
59
  - Datasets 3.1.0
60
+ - Tokenizers 0.20.3