Model save

README.md

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
-license: cc-by-nc-3.0
-base_model: Organika/sdxl-detector
+license: apache-2.0
+base_model: microsoft/swin-tiny-patch4-window7-224
 tags:
 - generated_from_trainer
 datasets:
@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.9512195121951219
+      value: 0.8870967741935484
 ---
 
 <!-- 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 [Organika/sdxl-detector](https://huggingface.co/Organika/sdxl-detector) on the imagefolder dataset.
+This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0688
-- Accuracy: 0.9512
+- Loss: 0.2812
+- Accuracy: 0.8871
 
 ## Model description
 
@@ -59,14 +59,14 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 3
+- num_epochs: 1
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| 0.0004        | 2.0833 | 50   | 0.0688          | 0.9512   |
+| 0.3707        | 0.5682 | 25   | 0.2812          | 0.8871   |
 
 
 ### Framework versions

config.json

@@ -1,6 +1,5 @@
 {
-  "_name_or_path": "Organika/sdxl-detector",
-  "_num_labels": 2,
+  "_name_or_path": "microsoft/swin-tiny-patch4-window7-224",
   "architectures": [
     "SwinForImageClassification"
   ],
@@ -8,35 +7,34 @@
   "depths": [
     2,
     2,
-    18,
+    6,
     2
   ],
   "drop_path_rate": 0.1,
-  "embed_dim": 128,
+  "embed_dim": 96,
   "encoder_stride": 32,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 1024,
+  "hidden_size": 768,
   "id2label": {
-    "0": "artificial",
-    "1": "human"
+    "0": "AI",
+    "1": "Real"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "artificial": 0,
-    "human": 1
+    "AI": "0",
+    "Real": "1"
   },
   "layer_norm_eps": 1e-05,
-  "max_length": 128,
   "mlp_ratio": 4.0,
   "model_type": "swin",
   "num_channels": 3,
   "num_heads": [
-    4,
-    8,
-    16,
-    32
+    3,
+    6,
+    12,
+    24
   ],
   "num_layers": 4,
   "out_features": [
@@ -45,7 +43,6 @@
   "out_indices": [
     4
   ],
-  "padding": "max_length",
   "patch_size": 4,
   "path_norm": true,
   "problem_type": "single_label_classification",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1c0d5afa003852c9e93880ca1baf5ebee5fc86a57f4ce8e517fca7d5664ea23b
-size 347498816
+oid sha256:37d687a35771f6f347ccd91a4bbc2c234a292986ecde55cc75988d8d5d8b52a8
+size 110342832

runs/Jan15_19-48-23_aa149997962b/events.out.tfevents.1736970512.aa149997962b.688.9

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

runs/Jan15_19-50-06_aa149997962b/events.out.tfevents.1736970612.aa149997962b.688.10

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

training_args.bin

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