Model save

README.md

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
-license: cc-by-4.0
-base_model: umm-maybe/AI-image-detector
+license: cc-by-nc-3.0
+base_model: Organika/sdxl-detector
 tags:
 - generated_from_trainer
 datasets:
@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 1.0
+      value: 0.9752066115702479
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,10 +31,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # finetuned-ai-real-swin
 
-This model is a fine-tuned version of [umm-maybe/AI-image-detector](https://huggingface.co/umm-maybe/AI-image-detector) on the imagefolder dataset.
+This model is a fine-tuned version of [Organika/sdxl-detector](https://huggingface.co/Organika/sdxl-detector) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0001
-- Accuracy: 1.0
+- Loss: 0.2485
+- Accuracy: 0.9752
 
 ## Model description
 
@@ -66,7 +66,7 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| 0.0009        | 2.2727 | 50   | 0.0001          | 1.0      |
+| 0.0141        | 2.2727 | 50   | 0.2485          | 0.9752   |
 
 
 ### Framework versions

all_results.json

@@ -1,13 +1,13 @@
 {
-    "epoch": 5.0,
-    "eval_accuracy": 0.9586776859504132,
-    "eval_loss": 0.22044861316680908,
-    "eval_runtime": 1.7849,
-    "eval_samples_per_second": 67.79,
-    "eval_steps_per_second": 8.964,
-    "total_flos": 2.663748891353088e+17,
-    "train_loss": 0.05316784412346103,
-    "train_runtime": 106.9013,
-    "train_samples_per_second": 31.805,
-    "train_steps_per_second": 1.029
+    "epoch": 3.0,
+    "eval_accuracy": 1.0,
+    "eval_loss": 0.000129295774968341,
+    "eval_runtime": 3.1734,
+    "eval_samples_per_second": 38.129,
+    "eval_steps_per_second": 5.042,
+    "total_flos": 1.5982493348118528e+17,
+    "train_loss": 0.07911565561186183,
+    "train_runtime": 65.9385,
+    "train_samples_per_second": 30.938,
+    "train_steps_per_second": 1.001
 }

config.json

@@ -1,5 +1,6 @@
 {
-  "_name_or_path": "umm-maybe/AI-image-detector",
+  "_name_or_path": "Organika/sdxl-detector",
+  "_num_labels": 2,
   "architectures": [
     "SwinForImageClassification"
   ],
@@ -23,8 +24,8 @@
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "artificial": "0",
-    "human": "1"
+    "artificial": 0,
+    "human": 1
   },
   "layer_norm_eps": 1e-05,
   "max_length": 128,

eval_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 5.0,
-    "eval_accuracy": 0.9586776859504132,
-    "eval_loss": 0.22044861316680908,
-    "eval_runtime": 1.7849,
-    "eval_samples_per_second": 67.79,
-    "eval_steps_per_second": 8.964
+    "epoch": 3.0,
+    "eval_accuracy": 1.0,
+    "eval_loss": 0.000129295774968341,
+    "eval_runtime": 3.1734,
+    "eval_samples_per_second": 38.129,
+    "eval_steps_per_second": 5.042
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:56d3f70bc13428ce7fcf6fc08ebcd51f6a472e32692d2b0534a7f0b3d14c2990
+oid sha256:229caa9409e25ff1f3b2271f08fbe14e69b015b83d5549a2028075de87b318d4
 size 347498816

runs/Dec31_08-31-31_3dcd8bc8b582/events.out.tfevents.1735633989.3dcd8bc8b582.14185.1

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

runs/Dec31_08-40-59_3dcd8bc8b582/events.out.tfevents.1735634464.3dcd8bc8b582.14185.2

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

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 5.0,
-    "total_flos": 2.663748891353088e+17,
-    "train_loss": 0.05316784412346103,
-    "train_runtime": 106.9013,
-    "train_samples_per_second": 31.805,
-    "train_steps_per_second": 1.029
+    "epoch": 3.0,
+    "total_flos": 1.5982493348118528e+17,
+    "train_loss": 0.07911565561186183,
+    "train_runtime": 65.9385,
+    "train_samples_per_second": 30.938,
+    "train_steps_per_second": 1.001
 }

trainer_state.json

@@ -1,122 +1,78 @@
 {
-  "best_metric": 0.1993160843849182,
+  "best_metric": 8.1218960986007e-05,
   "best_model_checkpoint": "finetuned-ai-real-swin/checkpoint-50",
-  "epoch": 5.0,
+  "epoch": 3.0,
   "eval_steps": 50,
-  "global_step": 110,
+  "global_step": 66,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
       "epoch": 0.45454545454545453,
-      "grad_norm": 0.0013060887577012181,
-      "learning_rate": 0.00018545454545454545,
-      "loss": 0.2915,
+      "grad_norm": 15.097875595092773,
+      "learning_rate": 0.00017272727272727275,
+      "loss": 0.2428,
       "step": 10
     },
     {
       "epoch": 0.9090909090909091,
-      "grad_norm": 0.018985752016305923,
-      "learning_rate": 0.00016727272727272728,
-      "loss": 0.0872,
+      "grad_norm": 2.583950996398926,
+      "learning_rate": 0.00014242424242424243,
+      "loss": 0.032,
       "step": 20
     },
     {
       "epoch": 1.3636363636363638,
-      "grad_norm": 3.094077328569256e-05,
-      "learning_rate": 0.0001490909090909091,
-      "loss": 0.0103,
+      "grad_norm": 0.0005581710720434785,
+      "learning_rate": 0.0001181818181818182,
+      "loss": 0.0741,
       "step": 30
     },
     {
       "epoch": 1.8181818181818183,
-      "grad_norm": 1.7741825580596924,
-      "learning_rate": 0.00013090909090909093,
-      "loss": 0.0615,
+      "grad_norm": 0.25307825207710266,
+      "learning_rate": 8.787878787878789e-05,
+      "loss": 0.1721,
       "step": 40
     },
     {
       "epoch": 2.2727272727272725,
-      "grad_norm": 0.0014310627011582255,
-      "learning_rate": 0.00011272727272727272,
-      "loss": 0.0133,
+      "grad_norm": 0.11153484135866165,
+      "learning_rate": 5.757575757575758e-05,
+      "loss": 0.0009,
       "step": 50
     },
     {
       "epoch": 2.2727272727272725,
-      "eval_accuracy": 0.9834710743801653,
-      "eval_loss": 0.1993160843849182,
-      "eval_runtime": 1.6646,
-      "eval_samples_per_second": 72.69,
-      "eval_steps_per_second": 9.612,
+      "eval_accuracy": 1.0,
+      "eval_loss": 8.1218960986007e-05,
+      "eval_runtime": 2.2287,
+      "eval_samples_per_second": 54.292,
+      "eval_steps_per_second": 7.179,
       "step": 50
     },
     {
       "epoch": 2.7272727272727275,
-      "grad_norm": 0.028763145208358765,
-      "learning_rate": 9.454545454545455e-05,
-      "loss": 0.0267,
+      "grad_norm": 0.003770518582314253,
+      "learning_rate": 2.7272727272727273e-05,
+      "loss": 0.0002,
       "step": 60
     },
     {
-      "epoch": 3.1818181818181817,
-      "grad_norm": 0.3835113048553467,
-      "learning_rate": 7.636363636363637e-05,
-      "loss": 0.0458,
-      "step": 70
-    },
-    {
-      "epoch": 3.6363636363636362,
-      "grad_norm": 0.016482913866639137,
-      "learning_rate": 5.818181818181818e-05,
-      "loss": 0.0047,
-      "step": 80
-    },
-    {
-      "epoch": 4.090909090909091,
-      "grad_norm": 0.0009892478119581938,
-      "learning_rate": 4e-05,
-      "loss": 0.0001,
-      "step": 90
-    },
-    {
-      "epoch": 4.545454545454545,
-      "grad_norm": 0.0006707807769998908,
-      "learning_rate": 2.1818181818181818e-05,
-      "loss": 0.0007,
-      "step": 100
-    },
-    {
-      "epoch": 4.545454545454545,
-      "eval_accuracy": 0.9586776859504132,
-      "eval_loss": 0.25598978996276855,
-      "eval_runtime": 2.0569,
-      "eval_samples_per_second": 58.827,
-      "eval_steps_per_second": 7.779,
-      "step": 100
-    },
-    {
-      "epoch": 5.0,
-      "grad_norm": 13.271048545837402,
-      "learning_rate": 3.636363636363636e-06,
-      "loss": 0.0432,
-      "step": 110
-    },
-    {
-      "epoch": 5.0,
-      "step": 110,
-      "total_flos": 2.663748891353088e+17,
-      "train_loss": 0.05316784412346103,
-      "train_runtime": 106.9013,
-      "train_samples_per_second": 31.805,
-      "train_steps_per_second": 1.029
+      "epoch": 3.0,
+      "step": 66,
+      "total_flos": 1.5982493348118528e+17,
+      "train_loss": 0.07911565561186183,
+      "train_runtime": 65.9385,
+      "train_samples_per_second": 30.938,
+      "train_steps_per_second": 1.001
     }
   ],
   "logging_steps": 10,
-  "max_steps": 110,
+  "max_steps": 66,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 5,
+  "num_train_epochs": 3,
   "save_steps": 100,
   "stateful_callbacks": {
     "TrainerControl": {
@@ -130,7 +86,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 2.663748891353088e+17,
+  "total_flos": 1.5982493348118528e+17,
   "train_batch_size": 32,
   "trial_name": null,
   "trial_params": null

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d2b65e74682131468fa2d5609238fbb268468f3dbf38eb1c751b2c6e50ee26fc
+oid sha256:aac1e3d03f13e22039e314aa587e786c0df1cabd5fbeb7624e362f11b696d660
 size 5304