Model save

README.md

@@ -16,8 +16,6 @@ should probably proofread and complete it, then remove this comment. -->
 # checkpoints
 
 This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 1.0604
 
 ## Model description
 
@@ -48,13 +46,13 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 1.0
+- training_steps: 3
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss |
-|:-------------:|:-----:|:----:|:---------------:|
-| 1.1049        | 1.0   | 23   | 1.0604          |
+| Training Loss | Epoch  | Step | Validation Loss |
+|:-------------:|:------:|:----:|:---------------:|
+| No log        | 0.1304 | 3    | 1.6955          |
 
 
 ### Framework versions

trainer_log.jsonl

@@ -1,4 +1,2 @@
-{"current_steps": 10, "total_steps": 23, "loss": 1.361, "learning_rate": 1.4539904997395468e-05, "epoch": 0.43478260869565216, "percentage": 43.48, "elapsed_time": "0:00:52", "remaining_time": "0:01:07"}
-{"current_steps": 20, "total_steps": 23, "loss": 1.1049, "learning_rate": 1.0899347581163222e-06, "epoch": 0.8695652173913043, "percentage": 86.96, "elapsed_time": "0:01:42", "remaining_time": "0:00:15"}
-{"current_steps": 23, "total_steps": 23, "eval_loss": 1.060397982597351, "epoch": 1.0, "percentage": 100.0, "elapsed_time": "0:03:30", "remaining_time": "0:00:00"}
-{"current_steps": 23, "total_steps": 23, "epoch": 1.0, "percentage": 100.0, "elapsed_time": "0:05:07", "remaining_time": "0:00:00"}
+{"current_steps": 3, "total_steps": 3, "eval_loss": 1.6955417394638062, "epoch": 0.13043478260869565, "percentage": 100.0, "elapsed_time": "0:01:49", "remaining_time": "0:00:00"}
+{"current_steps": 3, "total_steps": 3, "epoch": 0.13043478260869565, "percentage": 100.0, "elapsed_time": "0:03:27", "remaining_time": "0:00:00"}