Hugging Face's logo

chauhoang
/
b0760710-5585-42a9-a2ee-78e5b9d3694f

PEFT
Safetensors
gemma2
axolotl
Generated from Trainer
Model card Files
xet
 Community
chauhoang commited on
Commit
0b1ec07
·
verified ·
1 Parent(s): 26a0dca

End of training

Browse files
Files changed (2) hide show
  1. README.md +11 -4
  2. adapter_model.bin +1 -1
README.md CHANGED
@@ -65,7 +65,7 @@ lora_model_dir: null
65
  lora_r: 8
66
  lora_target_linear: true
67
  lr_scheduler: cosine
68
- max_steps: 1
69
  micro_batch_size: 2
70
  mlflow_experiment_name: /tmp/7152f650ecb2903c_train_data.json
71
  model_type: AutoModelForCausalLM
@@ -90,7 +90,7 @@ wandb_name: 2f783681-0954-497d-858e-f6a9740d789d
90
  wandb_project: Gradients-On-Demand
91
  wandb_run: your_name
92
  wandb_runid: 2f783681-0954-497d-858e-f6a9740d789d
93
- warmup_steps: 1
94
  weight_decay: 0.0
95
  xformers_attention: null
96
 
@@ -101,6 +101,8 @@ xformers_attention: null
101
  # b0760710-5585-42a9-a2ee-78e5b9d3694f
102
 
103
  This model is a fine-tuned version of [UCLA-AGI/Gemma-2-9B-It-SPPO-Iter2](https://huggingface.co/UCLA-AGI/Gemma-2-9B-It-SPPO-Iter2) on the None dataset.
 
 
104
 
105
  ## Model description
106
 
@@ -127,14 +129,19 @@ The following hyperparameters were used during training:
127
  - total_train_batch_size: 8
128
  - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
129
  - lr_scheduler_type: cosine
130
- - lr_scheduler_warmup_steps: 2
131
- - training_steps: 1
132
 
133
  ### Training results
134
 
135
  | Training Loss | Epoch | Step | Validation Loss |
136
  |:-------------:|:------:|:----:|:---------------:|
137
  | No log | 0.0000 | 1 | 5.6276 |
 
 
 
 
 
138
 
139
 
140
  ### Framework versions
 
65
  lora_r: 8
66
  lora_target_linear: true
67
  lr_scheduler: cosine
68
+ max_steps: 50
69
  micro_batch_size: 2
70
  mlflow_experiment_name: /tmp/7152f650ecb2903c_train_data.json
71
  model_type: AutoModelForCausalLM
 
90
  wandb_project: Gradients-On-Demand
91
  wandb_run: your_name
92
  wandb_runid: 2f783681-0954-497d-858e-f6a9740d789d
93
+ warmup_steps: 10
94
  weight_decay: 0.0
95
  xformers_attention: null
96
 
 
101
  # b0760710-5585-42a9-a2ee-78e5b9d3694f
102
 
103
  This model is a fine-tuned version of [UCLA-AGI/Gemma-2-9B-It-SPPO-Iter2](https://huggingface.co/UCLA-AGI/Gemma-2-9B-It-SPPO-Iter2) on the None dataset.
104
+ It achieves the following results on the evaluation set:
105
+ - Loss: 2.3163
106
 
107
  ## Model description
108
 
 
129
  - total_train_batch_size: 8
130
  - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
131
  - lr_scheduler_type: cosine
132
+ - lr_scheduler_warmup_steps: 10
133
+ - training_steps: 50
134
 
135
  ### Training results
136
 
137
  | Training Loss | Epoch | Step | Validation Loss |
138
  |:-------------:|:------:|:----:|:---------------:|
139
  | No log | 0.0000 | 1 | 5.6276 |
140
+ | 4.9356 | 0.0002 | 10 | 4.1117 |
141
+ | 2.5219 | 0.0004 | 20 | 2.6617 |
142
+ | 2.5043 | 0.0007 | 30 | 2.4102 |
143
+ | 2.3131 | 0.0009 | 40 | 2.3335 |
144
+ | 2.3433 | 0.0011 | 50 | 2.3163 |
145
 
146
 
147
  ### Framework versions
adapter_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1ed786598cc063b945c3f6de1ed0cc212821e8e8a1703a42f215de266b76706a
3
  size 108247866
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe79ae6d0f50e45a1f197f7710ca78e34348924c88e758a409e3bd51e18a6300
3
  size 108247866