End of training

README.md

@@ -1,5 +1,7 @@
 ---
 library_name: transformers
+license: apache-2.0
+base_model: facebook/convnext-base-224
 tags:
 - generated_from_trainer
 datasets:
@@ -14,14 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # convnext_model
 
-This model was trained from scratch on the imagefolder dataset.
+This model is a fine-tuned version of [facebook/convnext-base-224](https://huggingface.co/facebook/convnext-base-224) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.2166
-- eval_model_preparation_time: 0.002
-- eval_runtime: 34.083
-- eval_samples_per_second: 11.736
-- eval_steps_per_second: 0.734
-- step: 0
+- Loss: 0.0713
 
 ## Model description
 
@@ -48,9 +45,18 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 3
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 0.1274        | 1.0   | 100  | 0.1036          |
+| 0.0302        | 2.0   | 200  | 0.0760          |
+| 0.0117        | 3.0   | 300  | 0.0713          |
+
+
 ### Framework versions
 
 - Transformers 4.46.3
-- Pytorch 2.5.1+cpu
+- Pytorch 2.5.1+cu124
 - Datasets 3.1.0
 - Tokenizers 0.20.3

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "./saved_convnext_model/",
+  "_name_or_path": "facebook/convnext-base-224",
   "architectures": [
     "ConvNextForImageClassification"
   ],
@@ -18,18 +18,18 @@
     1024
   ],
   "id2label": {
-    "0": "cyclone",
-    "1": "earthquake",
-    "2": "flood",
-    "3": "wildfire"
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "cyclone": 0,
-    "earthquake": 1,
-    "flood": 2,
-    "wildfire": 3
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2,
+    "LABEL_3": 3
   },
   "layer_norm_eps": 1e-12,
   "layer_scale_init_value": 1e-06,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:251b2fdbfddef72aae4760dfab6086784821febaca54c198d4e9472934aa9367
+oid sha256:8382e1d8690eb91008ab0a54271146fa0e1d5667ec713df8459ceae703b94b54
 size 350323416

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:00c128e7cea18823884425e4cc03894288fc30bea6a37e8cd7046ae89c1dd958
+oid sha256:0c0ccb75cc167f00b48f9293b0c5f7516cac2fca6c70bfe3437060d466c9ebe1
 size 5240