Model save

README.md

@@ -3,8 +3,6 @@ library_name: transformers
 license: apache-2.0
 base_model: timm/resnet18.a1_in1k
 tags:
-- image-classification
-- vision
 - generated_from_trainer
 metrics:
 - accuracy
@@ -18,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-beans
 
-This model is a fine-tuned version of [timm/resnet18.a1_in1k](https://huggingface.co/timm/resnet18.a1_in1k) on the beans dataset.
+This model is a fine-tuned version of [timm/resnet18.a1_in1k](https://huggingface.co/timm/resnet18.a1_in1k) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0332
-- Accuracy: 0.6767
+- Loss: 0.8550
+- Accuracy: 0.7895
 
 ## Model description
 
@@ -46,17 +44,27 @@ The following hyperparameters were used during training:
 - seed: 1337
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 5.0
+- num_epochs: 15.0
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.0881        | 1.0   | 130  | 1.0902          | 0.4135   |
-| 1.0716        | 2.0   | 260  | 1.0685          | 0.5038   |
-| 1.061         | 3.0   | 390  | 1.0459          | 0.6241   |
-| 1.0514        | 4.0   | 520  | 1.0407          | 0.6015   |
-| 1.05          | 5.0   | 650  | 1.0332          | 0.6767   |
+| Training Loss | Epoch | Step | Accuracy | Validation Loss |
+|:-------------:|:-----:|:----:|:--------:|:---------------:|
+| 1.0881        | 1.0   | 130  | 0.4135   | 1.0902          |
+| 1.0716        | 2.0   | 260  | 0.5038   | 1.0685          |
+| 1.061         | 3.0   | 390  | 0.6241   | 1.0459          |
+| 1.0514        | 4.0   | 520  | 0.6015   | 1.0407          |
+| 1.05          | 5.0   | 650  | 0.6767   | 1.0332          |
+| 1.0357        | 6.0   | 780  | 1.0109   | 0.6541          |
+| 1.0012        | 7.0   | 910  | 0.9815   | 0.7368          |
+| 0.9932        | 8.0   | 1040 | 0.9550   | 0.7669          |
+| 0.9748        | 9.0   | 1170 | 0.9409   | 0.7669          |
+| 0.9113        | 10.0  | 1300 | 0.9149   | 0.7820          |
+| 0.9255        | 11.0  | 1430 | 0.8906   | 0.7895          |
+| 0.8877        | 12.0  | 1560 | 0.8749   | 0.7895          |
+| 0.9032        | 13.0  | 1690 | 0.8699   | 0.7970          |
+| 0.9001        | 14.0  | 1820 | 0.8674   | 0.7820          |
+| 0.8842        | 15.0  | 1950 | 0.8550   | 0.7895          |
 
 
 ### Framework versions

runs/Nov01_15-31-14_ip-10-90-1-182/events.out.tfevents.1730475078.ip-10-90-1-182.137679.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b0805cd1d09f4f0347e7e5dbaa689332cd01066578a2d2a4acf3cf8ee899b5ea
-size 35850
+oid sha256:5aa9b117265f232449d6ec2f1e1f30e6adbb53c58760babceffbf7c9979591ae
+size 36527