Upload folder using huggingface_hub

checkpoint-4750/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-4750/model.safetensors

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

checkpoint-4750/optimizer.pt

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

checkpoint-4750/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-4750/rng_state.pth

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

checkpoint-4750/scheduler.pt

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

checkpoint-4750/trainer_state.json

@@ -0,0 +1,196 @@
+{
+  "best_metric": 2.9626526832580566,
+  "best_model_checkpoint": "house-plant-image-detection/checkpoint-4750",
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 4750,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7829090191434774,
+      "eval_loss": 3.274250030517578,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 431.1188,
+      "eval_samples_per_second": 23.506,
+      "eval_steps_per_second": 2.939,
+      "step": 475
+    },
+    {
+      "epoch": 1.0526315789473684,
+      "grad_norm": 1.5620105266571045,
+      "learning_rate": 6.329787234042553e-07,
+      "loss": 3.2644,
+      "step": 500
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.8028419182948491,
+      "eval_loss": 3.2079219818115234,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 429.2124,
+      "eval_samples_per_second": 23.611,
+      "eval_steps_per_second": 2.952,
+      "step": 950
+    },
+    {
+      "epoch": 2.1052631578947367,
+      "grad_norm": 1.4659667015075684,
+      "learning_rate": 5.585106382978723e-07,
+      "loss": 3.1878,
+      "step": 1000
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.8171501874876653,
+      "eval_loss": 3.1499454975128174,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 428.1214,
+      "eval_samples_per_second": 23.671,
+      "eval_steps_per_second": 2.959,
+      "step": 1425
+    },
+    {
+      "epoch": 3.1578947368421053,
+      "grad_norm": 1.603775143623352,
+      "learning_rate": 4.840425531914894e-07,
+      "loss": 3.118,
+      "step": 1500
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.8248470495362147,
+      "eval_loss": 3.099377393722534,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 426.1632,
+      "eval_samples_per_second": 23.78,
+      "eval_steps_per_second": 2.973,
+      "step": 1900
+    },
+    {
+      "epoch": 4.2105263157894735,
+      "grad_norm": 1.6411057710647583,
+      "learning_rate": 4.0957446808510637e-07,
+      "loss": 3.0627,
+      "step": 2000
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.8316558121176238,
+      "eval_loss": 3.0566673278808594,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 429.8749,
+      "eval_samples_per_second": 23.574,
+      "eval_steps_per_second": 2.947,
+      "step": 2375
+    },
+    {
+      "epoch": 5.2631578947368425,
+      "grad_norm": 1.6472663879394531,
+      "learning_rate": 3.3510638297872343e-07,
+      "loss": 3.0104,
+      "step": 2500
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.835800276297612,
+      "eval_loss": 3.0226495265960693,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 433.9596,
+      "eval_samples_per_second": 23.352,
+      "eval_steps_per_second": 2.92,
+      "step": 2850
+    },
+    {
+      "epoch": 6.315789473684211,
+      "grad_norm": 1.7341008186340332,
+      "learning_rate": 2.606382978723404e-07,
+      "loss": 2.9695,
+      "step": 3000
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.8404381290704559,
+      "eval_loss": 2.9962480068206787,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 432.5356,
+      "eval_samples_per_second": 23.429,
+      "eval_steps_per_second": 2.929,
+      "step": 3325
+    },
+    {
+      "epoch": 7.368421052631579,
+      "grad_norm": 1.8458765745162964,
+      "learning_rate": 1.8617021276595742e-07,
+      "loss": 2.9413,
+      "step": 3500
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.8428063943161634,
+      "eval_loss": 2.9772186279296875,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 425.4068,
+      "eval_samples_per_second": 23.822,
+      "eval_steps_per_second": 2.978,
+      "step": 3800
+    },
+    {
+      "epoch": 8.421052631578947,
+      "grad_norm": 1.627718448638916,
+      "learning_rate": 1.1170212765957447e-07,
+      "loss": 2.9192,
+      "step": 4000
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.8439905269390172,
+      "eval_loss": 2.9663496017456055,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 427.5243,
+      "eval_samples_per_second": 23.704,
+      "eval_steps_per_second": 2.964,
+      "step": 4275
+    },
+    {
+      "epoch": 9.473684210526315,
+      "grad_norm": 1.6587204933166504,
+      "learning_rate": 3.723404255319149e-08,
+      "loss": 2.9038,
+      "step": 4500
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.844582593250444,
+      "eval_loss": 2.9626526832580566,
+      "eval_model_preparation_time": 0.0054,
+      "eval_runtime": 426.6341,
+      "eval_samples_per_second": 23.753,
+      "eval_steps_per_second": 2.97,
+      "step": 4750
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 4750,
+  "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.1782757816533832e+19,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-4750/training_args.bin

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

model.safetensors

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

training_args.bin

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