Upload folder using huggingface_hub

checkpoint-4680/config.json

@@ -0,0 +1,122 @@
+{
+  "_name_or_path": "dima806/house-plant-image-detection",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "African Violet (Saintpaulia ionantha)",
+    "1": "Aloe Vera",
+    "2": "Anthurium (Anthurium andraeanum)",
+    "3": "Areca Palm (Dypsis lutescens)",
+    "4": "Asparagus Fern (Asparagus setaceus)",
+    "5": "Begonia (Begonia spp.)",
+    "6": "Bird of Paradise (Strelitzia reginae)",
+    "7": "Birds Nest Fern (Asplenium nidus)",
+    "8": "Boston Fern (Nephrolepis exaltata)",
+    "9": "Calathea",
+    "10": "Cast Iron Plant (Aspidistra elatior)",
+    "11": "Chinese Money Plant (Pilea peperomioides)",
+    "12": "Chinese evergreen (Aglaonema)",
+    "13": "Christmas Cactus (Schlumbergera bridgesii)",
+    "14": "Chrysanthemum",
+    "15": "Ctenanthe",
+    "16": "Daffodils (Narcissus spp.)",
+    "17": "Dracaena",
+    "18": "Dumb Cane (Dieffenbachia spp.)",
+    "19": "Elephant Ear (Alocasia spp.)",
+    "20": "English Ivy (Hedera helix)",
+    "21": "Hyacinth (Hyacinthus orientalis)",
+    "22": "Iron Cross begonia (Begonia masoniana)",
+    "23": "Jade plant (Crassula ovata)",
+    "24": "Kalanchoe",
+    "25": "Lilium (Hemerocallis)",
+    "26": "Lily of the valley (Convallaria majalis)",
+    "27": "Money Tree (Pachira aquatica)",
+    "28": "Monstera Deliciosa (Monstera deliciosa)",
+    "29": "Orchid",
+    "30": "Parlor Palm (Chamaedorea elegans)",
+    "31": "Peace lily",
+    "32": "Poinsettia (Euphorbia pulcherrima)",
+    "33": "Polka Dot Plant (Hypoestes phyllostachya)",
+    "34": "Ponytail Palm (Beaucarnea recurvata)",
+    "35": "Pothos (Ivy arum)",
+    "36": "Prayer Plant (Maranta leuconeura)",
+    "37": "Rattlesnake Plant (Calathea lancifolia)",
+    "38": "Rubber Plant (Ficus elastica)",
+    "39": "Sago Palm (Cycas revoluta)",
+    "40": "Schefflera",
+    "41": "Snake plant (Sanseviera)",
+    "42": "Tradescantia",
+    "43": "Tulip",
+    "44": "Venus Flytrap",
+    "45": "Yucca",
+    "46": "ZZ Plant (Zamioculcas zamiifolia)"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "African Violet (Saintpaulia ionantha)": 0,
+    "Aloe Vera": 1,
+    "Anthurium (Anthurium andraeanum)": 2,
+    "Areca Palm (Dypsis lutescens)": 3,
+    "Asparagus Fern (Asparagus setaceus)": 4,
+    "Begonia (Begonia spp.)": 5,
+    "Bird of Paradise (Strelitzia reginae)": 6,
+    "Birds Nest Fern (Asplenium nidus)": 7,
+    "Boston Fern (Nephrolepis exaltata)": 8,
+    "Calathea": 9,
+    "Cast Iron Plant (Aspidistra elatior)": 10,
+    "Chinese Money Plant (Pilea peperomioides)": 11,
+    "Chinese evergreen (Aglaonema)": 12,
+    "Christmas Cactus (Schlumbergera bridgesii)": 13,
+    "Chrysanthemum": 14,
+    "Ctenanthe": 15,
+    "Daffodils (Narcissus spp.)": 16,
+    "Dracaena": 17,
+    "Dumb Cane (Dieffenbachia spp.)": 18,
+    "Elephant Ear (Alocasia spp.)": 19,
+    "English Ivy (Hedera helix)": 20,
+    "Hyacinth (Hyacinthus orientalis)": 21,
+    "Iron Cross begonia (Begonia masoniana)": 22,
+    "Jade plant (Crassula ovata)": 23,
+    "Kalanchoe": 24,
+    "Lilium (Hemerocallis)": 25,
+    "Lily of the valley (Convallaria majalis)": 26,
+    "Money Tree (Pachira aquatica)": 27,
+    "Monstera Deliciosa (Monstera deliciosa)": 28,
+    "Orchid": 29,
+    "Parlor Palm (Chamaedorea elegans)": 30,
+    "Peace lily": 31,
+    "Poinsettia (Euphorbia pulcherrima)": 32,
+    "Polka Dot Plant (Hypoestes phyllostachya)": 33,
+    "Ponytail Palm (Beaucarnea recurvata)": 34,
+    "Pothos (Ivy arum)": 35,
+    "Prayer Plant (Maranta leuconeura)": 36,
+    "Rattlesnake Plant (Calathea lancifolia)": 37,
+    "Rubber Plant (Ficus elastica)": 38,
+    "Sago Palm (Cycas revoluta)": 39,
+    "Schefflera": 40,
+    "Snake plant (Sanseviera)": 41,
+    "Tradescantia": 42,
+    "Tulip": 43,
+    "Venus Flytrap": 44,
+    "Yucca": 45,
+    "ZZ Plant (Zamioculcas zamiifolia)": 46
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.44.2"
+}

checkpoint-4680/model.safetensors

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

checkpoint-4680/optimizer.pt

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

checkpoint-4680/preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

checkpoint-4680/rng_state.pth

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

checkpoint-4680/scheduler.pt

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

checkpoint-4680/trainer_state.json

@@ -0,0 +1,196 @@
+{
+  "best_metric": 3.3425400257110596,
+  "best_model_checkpoint": "house-plant-image-detection/checkpoint-4680",
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 4680,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.20735249924872282,
+      "eval_loss": 3.7243916988372803,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 416.7605,
+      "eval_samples_per_second": 23.954,
+      "eval_steps_per_second": 2.995,
+      "step": 468
+    },
+    {
+      "epoch": 1.0683760683760684,
+      "grad_norm": 1.424812912940979,
+      "learning_rate": 8.125269978401728e-07,
+      "loss": 3.7509,
+      "step": 500
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.3650205349093459,
+      "eval_loss": 3.650330066680908,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 415.1984,
+      "eval_samples_per_second": 24.044,
+      "eval_steps_per_second": 3.006,
+      "step": 936
+    },
+    {
+      "epoch": 2.1367521367521367,
+      "grad_norm": 1.4496433734893799,
+      "learning_rate": 7.153347732181425e-07,
+      "loss": 3.6632,
+      "step": 1000
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.507362516277672,
+      "eval_loss": 3.5817322731018066,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 408.2157,
+      "eval_samples_per_second": 24.455,
+      "eval_steps_per_second": 3.057,
+      "step": 1404
+    },
+    {
+      "epoch": 3.2051282051282053,
+      "grad_norm": 1.4192156791687012,
+      "learning_rate": 6.181425485961123e-07,
+      "loss": 3.5805,
+      "step": 1500
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.6101372332966042,
+      "eval_loss": 3.5205447673797607,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 409.7634,
+      "eval_samples_per_second": 24.363,
+      "eval_steps_per_second": 3.046,
+      "step": 1872
+    },
+    {
+      "epoch": 4.273504273504273,
+      "grad_norm": 1.5265206098556519,
+      "learning_rate": 5.20950323974082e-07,
+      "loss": 3.5052,
+      "step": 2000
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.6736451968346189,
+      "eval_loss": 3.466716766357422,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 423.0883,
+      "eval_samples_per_second": 23.596,
+      "eval_steps_per_second": 2.95,
+      "step": 2340
+    },
+    {
+      "epoch": 5.3418803418803416,
+      "grad_norm": 1.5765515565872192,
+      "learning_rate": 4.237580993520518e-07,
+      "loss": 3.4422,
+      "step": 2500
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.7151156966843634,
+      "eval_loss": 3.422731876373291,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 423.7393,
+      "eval_samples_per_second": 23.559,
+      "eval_steps_per_second": 2.945,
+      "step": 2808
+    },
+    {
+      "epoch": 6.410256410256411,
+      "grad_norm": 1.4684735536575317,
+      "learning_rate": 3.265658747300216e-07,
+      "loss": 3.3898,
+      "step": 3000
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.7370529900831413,
+      "eval_loss": 3.3875274658203125,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 425.9892,
+      "eval_samples_per_second": 23.435,
+      "eval_steps_per_second": 2.93,
+      "step": 3276
+    },
+    {
+      "epoch": 7.478632478632479,
+      "grad_norm": 1.5869113206863403,
+      "learning_rate": 2.2937365010799135e-07,
+      "loss": 3.3477,
+      "step": 3500
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.7496744465591506,
+      "eval_loss": 3.36271071434021,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 425.1997,
+      "eval_samples_per_second": 23.478,
+      "eval_steps_per_second": 2.935,
+      "step": 3744
+    },
+    {
+      "epoch": 8.547008547008547,
+      "grad_norm": 1.4942816495895386,
+      "learning_rate": 1.321814254859611e-07,
+      "loss": 3.3201,
+      "step": 4000
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.7563858559551238,
+      "eval_loss": 3.347600221633911,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 426.4217,
+      "eval_samples_per_second": 23.411,
+      "eval_steps_per_second": 2.927,
+      "step": 4212
+    },
+    {
+      "epoch": 9.615384615384615,
+      "grad_norm": 1.5854607820510864,
+      "learning_rate": 3.4989200863930885e-08,
+      "loss": 3.3009,
+      "step": 4500
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.7592907943503957,
+      "eval_loss": 3.3425400257110596,
+      "eval_model_preparation_time": 0.0053,
+      "eval_runtime": 419.2995,
+      "eval_samples_per_second": 23.809,
+      "eval_steps_per_second": 2.976,
+      "step": 4680
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 4680,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 10,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.160833051811156e+19,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-4680/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/house-plant-image-detection",
   "architectures": [
     "ViTForImageClassification"
   ],

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cd11d1a55bc10fdb0393b76add0f1f16901b2b351a22593776a654f977fcd44c
+oid sha256:f4a84832a33db328eea9857cc95bbff810f5b49dfc0365f46e052b530b6f7a4b
 size 343362404

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:532a5c4ecfced93f28a090b423b2dad17231b16d0ea743bdcb04b94b1dd880ea
+oid sha256:6e63920ba5ce21a92a47ac4454fc135f2117176a194a1a8d6687090b74121760
 size 5112