Model save

README.md

@@ -4,8 +4,6 @@ license: apache-2.0
 base_model: microsoft/resnet-50
 tags:
 - generated_from_trainer
-metrics:
-- accuracy
 model-index:
 - name: resnet-50_rice-leaf-disease-augmented-v2_fft
   results: []
@@ -17,9 +15,6 @@ should probably proofread and complete it, then remove this comment. -->
 # resnet-50_rice-leaf-disease-augmented-v2_fft
 
 This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.8547
-- Accuracy: 0.7202
 
 ## Model description
 
@@ -46,27 +41,7 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: cosine_with_restarts
 - lr_scheduler_warmup_ratio: 0.1
 - num_epochs: 15
-
-### Training results
-
-| Training Loss | Epoch | Step | Accuracy | Validation Loss |
-|:-------------:|:-----:|:----:|:--------:|:---------------:|
-| 2.0458        | 1.0   | 125  | 0.4077   | 1.9620          |
-| 1.8522        | 2.0   | 250  | 0.5298   | 1.7118          |
-| 1.6181        | 3.0   | 375  | 0.5476   | 1.5115          |
-| 1.4376        | 4.0   | 500  | 0.6042   | 1.3707          |
-| 1.3093        | 5.0   | 625  | 0.6220   | 1.2792          |
-| 1.2299        | 6.0   | 750  | 0.6369   | 1.2290          |
-| 1.1881        | 7.0   | 875  | 0.6399   | 1.2191          |
-| 1.1751        | 8.0   | 1000 | 0.6488   | 1.2102          |
-| 1.1347        | 9.0   | 1125 | 0.6429   | 1.1266          |
-| 1.0039        | 10.0  | 1250 | 0.6726   | 1.0349          |
-| 0.907         | 11.0  | 1375 | 0.6935   | 0.9689          |
-| 0.8387        | 12.0  | 1500 | 0.6905   | 0.9262          |
-| 0.791         | 13.0  | 1625 | 0.8775   | 0.7143          |
-| 0.7244        | 14.0  | 1750 | 0.8533   | 0.7113          |
-| 0.691         | 15.0  | 1875 | 0.8547   | 0.7202          |
-
+- mixed_precision_training: Native AMP
 
 ### Framework versions
 

all_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 15.0,
+    "total_flos": 2.54958083702784e+18,
+    "train_loss": 0.14709369303385417,
+    "train_runtime": 1218.7812,
+    "train_samples_per_second": 98.459,
+    "train_steps_per_second": 1.538
+}

config.json

@@ -48,7 +48,6 @@
   "out_indices": [
     4
   ],
-  "problem_type": "single_label_classification",
   "stage_names": [
     "stem",
     "stage1",

logs/events.out.tfevents.1740284504.a837d8c25e39.930.0

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69221b0c6c05d76fa272da6f15c5ecd58d3b844b2c522853fd2c1e7e4fd078fd
+oid sha256:839860602f3a191ee8ba132b71280186bb2246bff269f4c6b563f35969aaee8f
 size 94352152

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 15.0,
+    "total_flos": 2.54958083702784e+18,
+    "train_loss": 0.14709369303385417,
+    "train_runtime": 1218.7812,
+    "train_samples_per_second": 98.459,
+    "train_steps_per_second": 1.538
+}

trainer_state.json

@@ -1,9 +1,9 @@
 {
-  "best_metric": 0.9262328147888184,
-  "best_model_checkpoint": "./drive/Shareddrives/CS198-Drones/training_output/resnet-50_rice-leaf-disease-augmented-v2_fft/checkpoint-1500",
-  "epoch": 12.0,
+  "best_metric": 0.8533294200897217,
+  "best_model_checkpoint": "./drive/Shareddrives/CS198-Drones/training_output/resnet-50_rice-leaf-disease-augmented-v2_fft/checkpoint-1750",
+  "epoch": 15.0,
   "eval_steps": 500,
-  "global_step": 1500,
+  "global_step": 1875,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -199,6 +199,63 @@
       "eval_samples_per_second": 25.277,
       "eval_steps_per_second": 0.451,
       "step": 1500
+    },
+    {
+      "epoch": 13.0,
+      "grad_norm": 3.1170542240142822,
+      "learning_rate": 8.42870045002354e-06,
+      "loss": 0.791,
+      "step": 1625
+    },
+    {
+      "epoch": 13.0,
+      "eval_accuracy": 0.7142857142857143,
+      "eval_loss": 0.8774951696395874,
+      "eval_runtime": 13.2469,
+      "eval_samples_per_second": 25.364,
+      "eval_steps_per_second": 0.453,
+      "step": 1625
+    },
+    {
+      "epoch": 14.0,
+      "grad_norm": 4.423528671264648,
+      "learning_rate": 3.018018951921804e-06,
+      "loss": 0.7244,
+      "step": 1750
+    },
+    {
+      "epoch": 14.0,
+      "eval_accuracy": 0.7113095238095238,
+      "eval_loss": 0.8533294200897217,
+      "eval_runtime": 13.4659,
+      "eval_samples_per_second": 24.952,
+      "eval_steps_per_second": 0.446,
+      "step": 1750
+    },
+    {
+      "epoch": 15.0,
+      "grad_norm": 4.605622291564941,
+      "learning_rate": 0.0,
+      "loss": 0.691,
+      "step": 1875
+    },
+    {
+      "epoch": 15.0,
+      "eval_accuracy": 0.7202380952380952,
+      "eval_loss": 0.8547136187553406,
+      "eval_runtime": 15.5224,
+      "eval_samples_per_second": 21.646,
+      "eval_steps_per_second": 0.387,
+      "step": 1875
+    },
+    {
+      "epoch": 15.0,
+      "step": 1875,
+      "total_flos": 2.54958083702784e+18,
+      "train_loss": 0.14709369303385417,
+      "train_runtime": 1218.7812,
+      "train_samples_per_second": 98.459,
+      "train_steps_per_second": 1.538
     }
   ],
   "logging_steps": 500,
@@ -213,12 +270,12 @@
         "should_evaluate": false,
         "should_log": false,
         "should_save": true,
-        "should_training_stop": false
+        "should_training_stop": true
       },
       "attributes": {}
     }
   },
-  "total_flos": 2.039664669622272e+18,
+  "total_flos": 2.54958083702784e+18,
   "train_batch_size": 64,
   "trial_name": null,
   "trial_params": null

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5fe8da1dd89984c4c60084967fbb9185dfabe13f9424a6020a088883d7e857b4
-size 5432
+oid sha256:e30e870845cfba6b60d75025a4d11acfd12cbaa6fe9d0eccf8216ae6f2b622f5
+size 5496