added model files

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

config.json

@@ -0,0 +1,63 @@
+{
+  "_name_or_path": "ai_detector_v3/checkpoint-2279",
+  "architectures": [
+    "SwinForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "depths": [
+    2,
+    2,
+    18,
+    2
+  ],
+  "drop_path_rate": 0.1,
+  "embed_dim": 128,
+  "encoder_stride": 32,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 1024,
+  "id2label": {
+    "0": "ai_gen",
+    "1": "human"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "label2id": {
+    "ai_gen": "0",
+    "human": "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
+  ],
+  "num_layers": 4,
+  "out_features": [
+    "stage4"
+  ],
+  "out_indices": [
+    4
+  ],
+  "padding": "max_length",
+  "patch_size": 4,
+  "path_norm": true,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "stage_names": [
+    "stem",
+    "stage1",
+    "stage2",
+    "stage3",
+    "stage4"
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.37.2",
+  "use_absolute_embeddings": false,
+  "window_size": 7
+}

model.safetensors

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

optimizer.pt

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

preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

rng_state.pth

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

scheduler.pt

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

trainer_state.json

@@ -0,0 +1,378 @@
+{
+  "best_metric": 0.996,
+  "best_model_checkpoint": "AI_ImageClassification_SDXL/checkpoint-586",
+  "epoch": 0.9995735607675906,
+  "eval_steps": 500,
+  "global_step": 586,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.02,
+      "learning_rate": 2.840909090909091e-06,
+      "loss": 1.2947,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 5.681818181818182e-06,
+      "loss": 0.689,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 8.522727272727273e-06,
+      "loss": 0.2576,
+      "step": 30
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.1363636363636365e-05,
+      "loss": 0.23,
+      "step": 40
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.4204545454545456e-05,
+      "loss": 0.1137,
+      "step": 50
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.7045454545454546e-05,
+      "loss": 0.0554,
+      "step": 60
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9886363636363638e-05,
+      "loss": 0.059,
+      "step": 70
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.272727272727273e-05,
+      "loss": 0.0583,
+      "step": 80
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.5568181818181817e-05,
+      "loss": 0.0349,
+      "step": 90
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.8409090909090912e-05,
+      "loss": 0.0864,
+      "step": 100
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 3.125e-05,
+      "loss": 0.0145,
+      "step": 110
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 3.409090909090909e-05,
+      "loss": 0.0358,
+      "step": 120
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 3.6931818181818184e-05,
+      "loss": 0.0473,
+      "step": 130
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 3.9772727272727275e-05,
+      "loss": 0.0253,
+      "step": 140
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 4.261363636363637e-05,
+      "loss": 0.0574,
+      "step": 150
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 4.545454545454546e-05,
+      "loss": 0.0403,
+      "step": 160
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 4.829545454545455e-05,
+      "loss": 0.0481,
+      "step": 170
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 4.9873577749683945e-05,
+      "loss": 0.0246,
+      "step": 180
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 4.955752212389381e-05,
+      "loss": 0.0094,
+      "step": 190
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 4.924146649810367e-05,
+      "loss": 0.0413,
+      "step": 200
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 4.892541087231353e-05,
+      "loss": 0.0339,
+      "step": 210
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 4.860935524652339e-05,
+      "loss": 0.0523,
+      "step": 220
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 4.829329962073325e-05,
+      "loss": 0.0104,
+      "step": 230
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 4.797724399494311e-05,
+      "loss": 0.0309,
+      "step": 240
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 4.7661188369152975e-05,
+      "loss": 0.0248,
+      "step": 250
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 4.734513274336283e-05,
+      "loss": 0.0306,
+      "step": 260
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 4.70290771175727e-05,
+      "loss": 0.0162,
+      "step": 270
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 4.6713021491782554e-05,
+      "loss": 0.016,
+      "step": 280
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 4.639696586599241e-05,
+      "loss": 0.0103,
+      "step": 290
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 4.608091024020228e-05,
+      "loss": 0.0196,
+      "step": 300
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 4.576485461441214e-05,
+      "loss": 0.0264,
+      "step": 310
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 4.5448798988622e-05,
+      "loss": 0.0344,
+      "step": 320
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 4.5132743362831855e-05,
+      "loss": 0.036,
+      "step": 330
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 4.4816687737041726e-05,
+      "loss": 0.0082,
+      "step": 340
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 4.4500632111251584e-05,
+      "loss": 0.0149,
+      "step": 350
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 4.418457648546144e-05,
+      "loss": 0.0082,
+      "step": 360
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 4.3868520859671306e-05,
+      "loss": 0.0451,
+      "step": 370
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 4.355246523388117e-05,
+      "loss": 0.0055,
+      "step": 380
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 4.323640960809103e-05,
+      "loss": 0.0087,
+      "step": 390
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 4.2920353982300885e-05,
+      "loss": 0.0052,
+      "step": 400
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 4.260429835651075e-05,
+      "loss": 0.0033,
+      "step": 410
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 4.2288242730720607e-05,
+      "loss": 0.0148,
+      "step": 420
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 4.197218710493047e-05,
+      "loss": 0.0134,
+      "step": 430
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 4.165613147914033e-05,
+      "loss": 0.0125,
+      "step": 440
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 4.134007585335019e-05,
+      "loss": 0.0531,
+      "step": 450
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 4.102402022756005e-05,
+      "loss": 0.0059,
+      "step": 460
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 4.0707964601769914e-05,
+      "loss": 0.0059,
+      "step": 470
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 4.039190897597978e-05,
+      "loss": 0.0194,
+      "step": 480
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 4.0075853350189636e-05,
+      "loss": 0.0111,
+      "step": 490
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 3.9759797724399494e-05,
+      "loss": 0.0321,
+      "step": 500
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 3.944374209860936e-05,
+      "loss": 0.0261,
+      "step": 510
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 3.912768647281922e-05,
+      "loss": 0.0196,
+      "step": 520
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 3.881163084702908e-05,
+      "loss": 0.0038,
+      "step": 530
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 3.849557522123894e-05,
+      "loss": 0.0216,
+      "step": 540
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 3.81795195954488e-05,
+      "loss": 0.0047,
+      "step": 550
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 3.7863463969658666e-05,
+      "loss": 0.0095,
+      "step": 560
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 3.754740834386852e-05,
+      "loss": 0.0036,
+      "step": 570
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 3.723135271807838e-05,
+      "loss": 0.0025,
+      "step": 580
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.996,
+      "eval_loss": 0.015431854873895645,
+      "eval_runtime": 44.7221,
+      "eval_samples_per_second": 44.721,
+      "eval_steps_per_second": 2.795,
+      "step": 586
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 1758,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 2.9390551738299924e+18,
+  "train_batch_size": 16,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

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