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: 0.7380
 - Accuracy: 0.8195
+- Loss: 0.7380
 
 ## Model description
 
@@ -50,23 +48,23 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.0861        | 1.0   | 130  | 1.0881          | 0.4286   |
-| 1.0631        | 2.0   | 260  | 1.0597          | 0.5414   |
-| 1.0443        | 3.0   | 390  | 1.0225          | 0.6692   |
-| 1.0218        | 4.0   | 520  | 0.9960          | 0.6842   |
-| 1.0028        | 5.0   | 650  | 0.9568          | 0.7293   |
-| 0.9752        | 6.0   | 780  | 0.9279          | 0.7669   |
-| 0.924         | 7.0   | 910  | 0.8852          | 0.8045   |
-| 0.9179        | 8.0   | 1040 | 0.8505          | 0.8120   |
-| 0.9           | 9.0   | 1170 | 0.8328          | 0.8045   |
-| 0.8084        | 10.0  | 1300 | 0.8071          | 0.8421   |
-| 0.8306        | 11.0  | 1430 | 0.7760          | 0.8346   |
-| 0.8031        | 12.0  | 1560 | 0.7563          | 0.8346   |
-| 0.8138        | 13.0  | 1690 | 0.7534          | 0.8421   |
-| 0.8178        | 14.0  | 1820 | 0.7508          | 0.8271   |
-| 0.7901        | 15.0  | 1950 | 0.7380          | 0.8195   |
+| Training Loss | Epoch | Step | Accuracy | Validation Loss |
+|:-------------:|:-----:|:----:|:--------:|:---------------:|
+| 1.0861        | 1.0   | 130  | 0.4286   | 1.0881          |
+| 1.0631        | 2.0   | 260  | 0.5414   | 1.0597          |
+| 1.0443        | 3.0   | 390  | 0.6692   | 1.0225          |
+| 1.0218        | 4.0   | 520  | 0.6842   | 0.9960          |
+| 1.0028        | 5.0   | 650  | 0.7293   | 0.9568          |
+| 0.9752        | 6.0   | 780  | 0.7669   | 0.9279          |
+| 0.924         | 7.0   | 910  | 0.8045   | 0.8852          |
+| 0.9179        | 8.0   | 1040 | 0.8120   | 0.8505          |
+| 0.9           | 9.0   | 1170 | 0.8045   | 0.8328          |
+| 0.8084        | 10.0  | 1300 | 0.8421   | 0.8071          |
+| 0.8306        | 11.0  | 1430 | 0.8346   | 0.7760          |
+| 0.8031        | 12.0  | 1560 | 0.8346   | 0.7563          |
+| 0.8138        | 13.0  | 1690 | 0.8421   | 0.7534          |
+| 0.8178        | 14.0  | 1820 | 0.8271   | 0.7508          |
+| 0.7901        | 15.0  | 1950 | 0.8195   | 0.7380          |
 
 
 ### Framework versions

config.json

@@ -17,7 +17,7 @@
     "healthy": "2"
   },
   "model_type": "timm_wrapper",
-  "num_classes": 1000,
+  "num_classes": 3,
   "num_features": 512,
   "pretrained_cfg": {
     "classifier": "fc",
@@ -37,7 +37,6 @@
       0.456,
       0.406
     ],
-    "num_classes": 1000,
     "origin_url": "https://github.com/huggingface/pytorch-image-models",
     "paper_ids": "arXiv:2110.00476",
     "pool_size": [
@@ -57,7 +56,6 @@
       288
     ]
   },
-  "problem_type": "single_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.47.0.dev0"
 }

emissions.csv

@@ -1,2 +1,3 @@
 timestamp,experiment_id,project_name,duration,emissions,energy_consumed,country_name,country_iso_code,region,on_cloud,cloud_provider,cloud_region
 2024-11-19T21:30:14,755bbb60-c220-4645-b6ef-6052b62dc3d4,codecarbon,138.7698619365692,0.0017126449757406481,0.004639624695827094,United States,USA,virginia,N,,
+2024-11-19T22:38:10,1e99c1f1-c5cf-4dd1-a260-5aed068b8ccc,codecarbon,0.02003312110900879,1.0601304553280803e-07,2.871936397215943e-07,United States,USA,virginia,N,,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c9cecaf222446115c5a889128c3565ba323ee918bafb1f20740b5d866726c0ed
+oid sha256:2a306827aa454fce37d405b859666518e4035baf597549bdcaff00264f7a4293
 size 44761548

runs/Nov19_22-38-05_ip-10-90-1-182/events.out.tfevents.1732055888.ip-10-90-1-182.3352928.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2ae56c507c4fa7eccb4db7559e309d6de3119e6ebd6c063d65b5e253394d6c4
+size 5921

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:010ee6fd6e6bcde7a7bedce4a4ba5bb431cc40956d1174ca9782b2f8bb876b37
+oid sha256:d0432369fa9075060b2ca4bcc422e444da973cd81eebf8e7fb8560a47879656b
 size 5432