End of training

README.md

@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [codellama/CodeLlama-7b-hf](https://huggingface.co/codellama/CodeLlama-7b-hf) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4621
+- Loss: 0.5902
 
 ## Model description
 
@@ -39,37 +39,21 @@ The following hyperparameters were used during training:
 - train_batch_size: 32
 - eval_batch_size: 8
 - seed: 42
-- gradient_accumulation_steps: 4
-- total_train_batch_size: 128
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 64
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 100
-- training_steps: 400
+- training_steps: 80
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 2.1563        | 0.04  | 20   | 1.9767          |
-| 1.1222        | 0.07  | 40   | 0.8267          |
-| 0.814         | 0.11  | 60   | 0.6637          |
-| 0.5956        | 0.14  | 80   | 0.5908          |
-| 0.405         | 0.18  | 100  | 0.5643          |
-| 0.5643        | 0.22  | 120  | 0.5204          |
-| 0.4326        | 0.25  | 140  | 0.5107          |
-| 0.6401        | 0.29  | 160  | 0.5211          |
-| 0.4789        | 0.33  | 180  | 0.4908          |
-| 0.3577        | 0.36  | 200  | 0.5069          |
-| 0.5289        | 0.4   | 220  | 0.4851          |
-| 0.3971        | 0.43  | 240  | 0.4811          |
-| 0.5972        | 0.47  | 260  | 0.4807          |
-| 0.4683        | 0.51  | 280  | 0.4712          |
-| 0.3442        | 0.54  | 300  | 0.4790          |
-| 0.5148        | 0.58  | 320  | 0.4692          |
-| 0.3917        | 0.62  | 340  | 0.4661          |
-| 0.5769        | 0.65  | 360  | 0.4661          |
-| 0.4603        | 0.69  | 380  | 0.4629          |
-| 0.3461        | 0.72  | 400  | 0.4621          |
+| 2.1581        | 0.02  | 20   | 1.9857          |
+| 1.1128        | 0.04  | 40   | 0.8156          |
+| 0.8092        | 0.05  | 60   | 0.6632          |
+| 0.5965        | 0.07  | 80   | 0.5902          |
 
 
 ### Framework versions