🍻 cheers

README.md

@@ -3,6 +3,7 @@ library_name: transformers
 license: apache-2.0
 base_model: google/vit-base-patch16-224
 tags:
+- image-classification
 - generated_from_trainer
 metrics:
 - accuracy
@@ -16,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # realFake-img
 
-This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the ai_real_images dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.4633
 - Accuracy: 0.8836

all_results.json

@@ -1,13 +1,13 @@
 {
     "epoch": 4.0,
-    "eval_accuracy": 0.8424657534246576,
-    "eval_loss": 0.339973121881485,
-    "eval_runtime": 4.934,
-    "eval_samples_per_second": 29.591,
-    "eval_steps_per_second": 3.851,
+    "eval_accuracy": 0.8835616438356164,
+    "eval_loss": 0.46333640813827515,
+    "eval_runtime": 5.0188,
+    "eval_samples_per_second": 29.091,
+    "eval_steps_per_second": 3.786,
     "total_flos": 2.5832369176982323e+17,
-    "train_loss": 0.38543382860147035,
-    "train_runtime": 140.7513,
-    "train_samples_per_second": 23.474,
-    "train_steps_per_second": 1.478
+    "train_loss": 0.1598776997281955,
+    "train_runtime": 141.5375,
+    "train_samples_per_second": 23.344,
+    "train_steps_per_second": 1.47
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 4.0,
-    "eval_accuracy": 0.8424657534246576,
-    "eval_loss": 0.339973121881485,
-    "eval_runtime": 4.934,
-    "eval_samples_per_second": 29.591,
-    "eval_steps_per_second": 3.851
+    "eval_accuracy": 0.8835616438356164,
+    "eval_loss": 0.46333640813827515,
+    "eval_runtime": 5.0188,
+    "eval_samples_per_second": 29.091,
+    "eval_steps_per_second": 3.786
 }

runs/Sep24_13-50-15_37647b9c0688/events.out.tfevents.1727186424.37647b9c0688.3797.5

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

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 4.0,
     "total_flos": 2.5832369176982323e+17,
-    "train_loss": 0.38543382860147035,
-    "train_runtime": 140.7513,
-    "train_samples_per_second": 23.474,
-    "train_steps_per_second": 1.478
+    "train_loss": 0.1598776997281955,
+    "train_runtime": 141.5375,
+    "train_samples_per_second": 23.344,
+    "train_steps_per_second": 1.47
 }

trainer_state.json

@@ -1,5 +1,5 @@
 {
-  "best_metric": 0.339973121881485,
+  "best_metric": 0.46333640813827515,
   "best_model_checkpoint": "realFake-img/checkpoint-200",
   "epoch": 4.0,
   "eval_steps": 100,
@@ -10,170 +10,170 @@
   "log_history": [
     {
       "epoch": 0.19230769230769232,
-      "grad_norm": 3.016617774963379,
-      "learning_rate": 0.00019134615384615387,
-      "loss": 0.7043,
+      "grad_norm": 5.288677215576172,
+      "learning_rate": 0.00019038461538461538,
+      "loss": 0.1567,
       "step": 10
     },
     {
       "epoch": 0.38461538461538464,
-      "grad_norm": 8.659211158752441,
-      "learning_rate": 0.00018173076923076923,
-      "loss": 0.6331,
+      "grad_norm": 4.266819953918457,
+      "learning_rate": 0.00018076923076923077,
+      "loss": 0.2467,
       "step": 20
     },
     {
       "epoch": 0.5769230769230769,
-      "grad_norm": 3.7159981727600098,
-      "learning_rate": 0.00017211538461538463,
-      "loss": 0.6697,
+      "grad_norm": 3.213836193084717,
+      "learning_rate": 0.00017115384615384616,
+      "loss": 0.3028,
       "step": 30
     },
     {
       "epoch": 0.7692307692307693,
-      "grad_norm": 5.222133636474609,
-      "learning_rate": 0.00016250000000000002,
-      "loss": 0.5777,
+      "grad_norm": 0.7542410492897034,
+      "learning_rate": 0.00016153846153846155,
+      "loss": 0.1968,
       "step": 40
     },
     {
       "epoch": 0.9615384615384616,
-      "grad_norm": 6.754155158996582,
-      "learning_rate": 0.00015288461538461539,
-      "loss": 0.5347,
+      "grad_norm": 11.869466781616211,
+      "learning_rate": 0.00015192307692307692,
+      "loss": 0.2678,
       "step": 50
     },
     {
       "epoch": 1.1538461538461537,
-      "grad_norm": 1.658116102218628,
-      "learning_rate": 0.00014326923076923078,
-      "loss": 0.4377,
+      "grad_norm": 0.7070229649543762,
+      "learning_rate": 0.0001423076923076923,
+      "loss": 0.1494,
       "step": 60
     },
     {
       "epoch": 1.3461538461538463,
-      "grad_norm": 5.012947082519531,
-      "learning_rate": 0.00013365384615384614,
-      "loss": 0.4354,
+      "grad_norm": 3.979475975036621,
+      "learning_rate": 0.0001326923076923077,
+      "loss": 0.0927,
       "step": 70
     },
     {
       "epoch": 1.5384615384615383,
-      "grad_norm": 4.971016883850098,
-      "learning_rate": 0.00012403846153846154,
-      "loss": 0.4928,
+      "grad_norm": 6.856090545654297,
+      "learning_rate": 0.0001230769230769231,
+      "loss": 0.2443,
       "step": 80
     },
     {
       "epoch": 1.7307692307692308,
-      "grad_norm": 4.061134338378906,
-      "learning_rate": 0.00011442307692307692,
-      "loss": 0.3803,
+      "grad_norm": 0.7266550064086914,
+      "learning_rate": 0.00011346153846153846,
+      "loss": 0.1968,
       "step": 90
     },
     {
       "epoch": 1.9230769230769231,
-      "grad_norm": 4.557286739349365,
-      "learning_rate": 0.00010480769230769232,
-      "loss": 0.3732,
+      "grad_norm": 3.321963310241699,
+      "learning_rate": 0.00010384615384615386,
+      "loss": 0.1137,
       "step": 100
     },
     {
       "epoch": 1.9230769230769231,
-      "eval_accuracy": 0.7945205479452054,
-      "eval_loss": 0.4088120758533478,
-      "eval_runtime": 5.0993,
-      "eval_samples_per_second": 28.631,
-      "eval_steps_per_second": 3.726,
+      "eval_accuracy": 0.8287671232876712,
+      "eval_loss": 0.48689645528793335,
+      "eval_runtime": 5.0712,
+      "eval_samples_per_second": 28.79,
+      "eval_steps_per_second": 3.747,
       "step": 100
     },
     {
       "epoch": 2.1153846153846154,
-      "grad_norm": 2.305349826812744,
-      "learning_rate": 9.519230769230769e-05,
-      "loss": 0.3172,
+      "grad_norm": 6.398293972015381,
+      "learning_rate": 9.423076923076924e-05,
+      "loss": 0.0703,
       "step": 110
     },
     {
       "epoch": 2.3076923076923075,
-      "grad_norm": 1.443526029586792,
-      "learning_rate": 8.557692307692308e-05,
-      "loss": 0.3155,
+      "grad_norm": 0.38959336280822754,
+      "learning_rate": 8.461538461538461e-05,
+      "loss": 0.1407,
       "step": 120
     },
     {
       "epoch": 2.5,
-      "grad_norm": 1.3094813823699951,
-      "learning_rate": 7.596153846153846e-05,
-      "loss": 0.2671,
+      "grad_norm": 0.1938396394252777,
+      "learning_rate": 7.500000000000001e-05,
+      "loss": 0.1098,
       "step": 130
     },
     {
       "epoch": 2.6923076923076925,
-      "grad_norm": 5.578798770904541,
-      "learning_rate": 6.634615384615385e-05,
-      "loss": 0.2603,
+      "grad_norm": 5.93131685256958,
+      "learning_rate": 6.538461538461539e-05,
+      "loss": 0.0969,
       "step": 140
     },
     {
       "epoch": 2.8846153846153846,
-      "grad_norm": 1.8392003774642944,
-      "learning_rate": 5.673076923076923e-05,
-      "loss": 0.2804,
+      "grad_norm": 0.0541561059653759,
+      "learning_rate": 5.576923076923077e-05,
+      "loss": 0.1299,
       "step": 150
     },
     {
       "epoch": 3.076923076923077,
-      "grad_norm": 0.6986391544342041,
-      "learning_rate": 4.711538461538462e-05,
-      "loss": 0.3171,
+      "grad_norm": 3.552217960357666,
+      "learning_rate": 4.615384615384616e-05,
+      "loss": 0.1264,
       "step": 160
     },
     {
       "epoch": 3.269230769230769,
-      "grad_norm": 4.677719593048096,
-      "learning_rate": 3.7500000000000003e-05,
-      "loss": 0.1962,
+      "grad_norm": 8.6173095703125,
+      "learning_rate": 3.653846153846154e-05,
+      "loss": 0.1076,
       "step": 170
     },
     {
       "epoch": 3.4615384615384617,
-      "grad_norm": 1.8367177248001099,
-      "learning_rate": 2.7884615384615386e-05,
-      "loss": 0.3216,
+      "grad_norm": 0.7253573536872864,
+      "learning_rate": 2.6923076923076923e-05,
+      "loss": 0.157,
       "step": 180
     },
     {
       "epoch": 3.6538461538461537,
-      "grad_norm": 5.29152774810791,
-      "learning_rate": 1.826923076923077e-05,
-      "loss": 0.1938,
+      "grad_norm": 7.486116409301758,
+      "learning_rate": 1.730769230769231e-05,
+      "loss": 0.177,
       "step": 190
     },
     {
       "epoch": 3.8461538461538463,
-      "grad_norm": 1.2080388069152832,
-      "learning_rate": 8.653846153846155e-06,
-      "loss": 0.1451,
+      "grad_norm": 1.391594409942627,
+      "learning_rate": 7.692307692307694e-06,
+      "loss": 0.1002,
       "step": 200
     },
     {
       "epoch": 3.8461538461538463,
-      "eval_accuracy": 0.8424657534246576,
-      "eval_loss": 0.339973121881485,
-      "eval_runtime": 4.8463,
-      "eval_samples_per_second": 30.126,
-      "eval_steps_per_second": 3.921,
+      "eval_accuracy": 0.8835616438356164,
+      "eval_loss": 0.46333640813827515,
+      "eval_runtime": 4.8332,
+      "eval_samples_per_second": 30.208,
+      "eval_steps_per_second": 3.931,
       "step": 200
     },
     {
       "epoch": 4.0,
       "step": 208,
       "total_flos": 2.5832369176982323e+17,
-      "train_loss": 0.38543382860147035,
-      "train_runtime": 140.7513,
-      "train_samples_per_second": 23.474,
-      "train_steps_per_second": 1.478
+      "train_loss": 0.1598776997281955,
+      "train_runtime": 141.5375,
+      "train_samples_per_second": 23.344,
+      "train_steps_per_second": 1.47
     }
   ],
   "logging_steps": 10,