Model save

README.md

@@ -14,12 +14,14 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
+[<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/jalitv77-ml/lora_transformers/runs/q00oyrrm)
+[<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/jalitv77-ml/lora_transformers/runs/q00oyrrm)
 # vit-base-patch16-224-in21k-finetuned-lora-food101
 
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5269
-- Accuracy: 0.8574
+- Loss: 0.5198
+- Accuracy: 0.8565
 
 ## Model description
 
@@ -53,11 +55,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| 0.8173        | 0.9981 | 133  | 0.6705          | 0.8170   |
-| 0.7385        | 1.9962 | 266  | 0.6100          | 0.8313   |
-| 0.5756        | 2.9944 | 399  | 0.5643          | 0.8471   |
-| 0.5668        | 4.0    | 533  | 0.5368          | 0.8527   |
-| 0.5           | 4.9906 | 665  | 0.5269          | 0.8574   |
+| 0.8397        | 0.9981 | 133  | 0.6682          | 0.8154   |
+| 0.8471        | 1.9962 | 266  | 0.5984          | 0.8366   |
+| 0.6114        | 2.9944 | 399  | 0.5590          | 0.8438   |
+| 0.6202        | 4.0    | 533  | 0.5335          | 0.8532   |
+| 0.4775        | 4.9906 | 665  | 0.5198          | 0.8565   |
 
 
 ### Framework versions