Jiyog/fine-tuned-kitchenobj-resnet50

README.md

@@ -1,5 +1,7 @@
 ---
 library_name: transformers
+license: apache-2.0
+base_model: microsoft/resnet-50
 tags:
 - generated_from_trainer
 datasets:
@@ -22,10 +24,10 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.6008750607681089
+      value: 0.8677685950413223
     - name: F1
       type: f1
-      value: 0.5946155383964072
+      value: 0.8678765100301015
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,11 +35,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # resnet-kitchen-object
 
-This model is a fine-tuned version of [](https://huggingface.co/) on the imagefolder dataset.
+This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.2913
-- Accuracy: 0.6009
-- F1: 0.5946
+- Loss: 0.4314
+- Accuracy: 0.8678
+- F1: 0.8679
 
 ## Model description
 
@@ -56,34 +58,32 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0003
-- train_batch_size: 32
-- eval_batch_size: 32
+- learning_rate: 5e-05
+- train_batch_size: 16
+- eval_batch_size: 16
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.05
-- num_epochs: 14
+- num_epochs: 12
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| 2.3508        | 1.0   | 224  | 3.0948          | 0.1779   | 0.1492 |
-| 2.0324        | 2.0   | 448  | 2.0692          | 0.2615   | 0.2454 |
-| 1.8579        | 3.0   | 672  | 2.0103          | 0.2528   | 0.2369 |
-| 1.7853        | 4.0   | 896  | 3.5934          | 0.2470   | 0.2224 |
-| 1.666         | 5.0   | 1120 | 2.0879          | 0.3175   | 0.3163 |
-| 1.5695        | 6.0   | 1344 | 1.6535          | 0.4176   | 0.4115 |
-| 1.4633        | 7.0   | 1568 | 1.5123          | 0.4754   | 0.4698 |
-| 1.3519        | 8.0   | 1792 | 1.4649          | 0.5070   | 0.4970 |
-| 1.2921        | 9.0   | 2016 | 1.8045          | 0.4404   | 0.4389 |
-| 1.1215        | 10.0  | 2240 | 1.3583          | 0.5659   | 0.5654 |
-| 1.0105        | 11.0  | 2464 | 1.3541          | 0.5639   | 0.5551 |
-| 0.8996        | 12.0  | 2688 | 1.2972          | 0.5800   | 0.5777 |
-| 0.7803        | 13.0  | 2912 | 1.2917          | 0.5955   | 0.5913 |
-| 0.7978        | 14.0  | 3136 | 1.2913          | 0.6009   | 0.5946 |
+| 2.0478        | 1.0   | 447  | 1.8372          | 0.5119   | 0.5147 |
+| 0.9774        | 2.0   | 894  | 0.7962          | 0.7832   | 0.7816 |
+| 0.6756        | 3.0   | 1341 | 0.5893          | 0.8221   | 0.8211 |
+| 0.5692        | 4.0   | 1788 | 0.5361          | 0.8347   | 0.8349 |
+| 0.5087        | 5.0   | 2235 | 0.5034          | 0.8439   | 0.8438 |
+| 0.4525        | 6.0   | 2682 | 0.4738          | 0.8483   | 0.8480 |
+| 0.4211        | 7.0   | 3129 | 0.4518          | 0.8610   | 0.8604 |
+| 0.4156        | 8.0   | 3576 | 0.4418          | 0.8629   | 0.8628 |
+| 0.3394        | 9.0   | 4023 | 0.4394          | 0.8663   | 0.8659 |
+| 0.3452        | 10.0  | 4470 | 0.4341          | 0.8653   | 0.8654 |
+| 0.3121        | 11.0  | 4917 | 0.4457          | 0.8658   | 0.8655 |
+| 0.3392        | 12.0  | 5364 | 0.4314          | 0.8678   | 0.8679 |
 
 
 ### Framework versions

config.json

@@ -20,29 +20,28 @@
     2048
   ],
   "id2label": {
-    "0": "bowl",
-    "1": "fork",
-    "2": "glass",
-    "3": "knife",
-    "4": "mug",
-    "5": "pan",
-    "6": "plate",
-    "7": "spatula",
-    "8": "spoon",
-    "9": "whisk"
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3",
+    "4": "LABEL_4",
+    "5": "LABEL_5",
+    "6": "LABEL_6",
+    "7": "LABEL_7",
+    "8": "LABEL_8",
+    "9": "LABEL_9"
   },
-  "image_size": 224,
   "label2id": {
-    "bowl": "0",
-    "fork": "1",
-    "glass": "2",
-    "knife": "3",
-    "mug": "4",
-    "pan": "5",
-    "plate": "6",
-    "spatula": "7",
-    "spoon": "8",
-    "whisk": "9"
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2,
+    "LABEL_3": 3,
+    "LABEL_4": 4,
+    "LABEL_5": 5,
+    "LABEL_6": 6,
+    "LABEL_7": 7,
+    "LABEL_8": 8,
+    "LABEL_9": 9
   },
   "layer_type": "bottleneck",
   "model_type": "resnet",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2eaedf7eca1ef49907e4c8dcc006595ee5b3998380335743534759702dd330d3
+oid sha256:dbb9013a4e54bdbb351c7a49d95bbccf3b196e1e531d0192a24172fd614add05
 size 94368544

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d0483edade651a6423e1dd72b6117037de4da1b9cd0a1a12b75ebd593faba568
+oid sha256:71d93c0305244e2f0e7405e0dd1665c029588f619eb5eb06e234666bd6be6b77
 size 5841