sooks commited on
Commit
d34add7
1 Parent(s): a6dbf35

Model save

Browse files
README.md ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: facebook/convnext-tiny-224
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ model-index:
9
+ - name: idbwtiny
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # idbwtiny
17
+
18
+ This model is a fine-tuned version of [facebook/convnext-tiny-224](https://huggingface.co/facebook/convnext-tiny-224) on an unknown dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.1267
21
+ - Accuracy: 0.9569
22
+
23
+ ## Model description
24
+
25
+ More information needed
26
+
27
+ ## Intended uses & limitations
28
+
29
+ More information needed
30
+
31
+ ## Training and evaluation data
32
+
33
+ More information needed
34
+
35
+ ## Training procedure
36
+
37
+ ### Training hyperparameters
38
+
39
+ The following hyperparameters were used during training:
40
+ - learning_rate: 0.0002
41
+ - train_batch_size: 150
42
+ - eval_batch_size: 8
43
+ - seed: 42
44
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
+ - lr_scheduler_type: linear
46
+ - num_epochs: 10
47
+ - mixed_precision_training: Native AMP
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|
53
+ | 0.3871 | 1.0 | 2010 | 0.3021 | 0.8674 |
54
+ | 0.3403 | 2.0 | 4020 | 0.2442 | 0.8967 |
55
+ | 0.3058 | 3.0 | 6030 | 0.2269 | 0.9052 |
56
+ | 0.2722 | 4.0 | 8040 | 0.2008 | 0.9197 |
57
+ | 0.2496 | 5.0 | 10050 | 0.1841 | 0.9272 |
58
+ | 0.2219 | 6.0 | 12060 | 0.1706 | 0.9353 |
59
+ | 0.198 | 7.0 | 14070 | 0.1478 | 0.9452 |
60
+ | 0.1812 | 8.0 | 16080 | 0.1413 | 0.9503 |
61
+ | 0.1632 | 9.0 | 18090 | 0.1301 | 0.9553 |
62
+ | 0.1576 | 10.0 | 20100 | 0.1267 | 0.9569 |
63
+
64
+
65
+ ### Framework versions
66
+
67
+ - Transformers 4.38.2
68
+ - Pytorch 2.2.1+cu121
69
+ - Datasets 2.19.0
70
+ - Tokenizers 0.15.2
config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/convnext-tiny-224",
3
+ "architectures": [
4
+ "ConvNextForImageClassification"
5
+ ],
6
+ "depths": [
7
+ 3,
8
+ 3,
9
+ 9,
10
+ 3
11
+ ],
12
+ "drop_path_rate": 0.0,
13
+ "hidden_act": "gelu",
14
+ "hidden_sizes": [
15
+ 96,
16
+ 192,
17
+ 384,
18
+ 768
19
+ ],
20
+ "id2label": {
21
+ "0": "AI",
22
+ "1": "Human"
23
+ },
24
+ "image_size": 224,
25
+ "initializer_range": 0.02,
26
+ "label2id": {
27
+ "AI": "0",
28
+ "Human": "1"
29
+ },
30
+ "layer_norm_eps": 1e-12,
31
+ "layer_scale_init_value": 1e-06,
32
+ "model_type": "convnext",
33
+ "num_channels": 3,
34
+ "num_stages": 4,
35
+ "out_features": [
36
+ "stage4"
37
+ ],
38
+ "out_indices": [
39
+ 4
40
+ ],
41
+ "patch_size": 4,
42
+ "problem_type": "single_label_classification",
43
+ "stage_names": [
44
+ "stem",
45
+ "stage1",
46
+ "stage2",
47
+ "stage3",
48
+ "stage4"
49
+ ],
50
+ "torch_dtype": "float32",
51
+ "transformers_version": "4.38.2"
52
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c9f2abab038e9743a534599490227606473edbb30ddd1f3acd8c4d5da3e6c7f
3
+ size 111307920
preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_pct": 0.875,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "image_processor_type": "ConvNextFeatureExtractor",
12
+ "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
+ ],
17
+ "resample": 3,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": [
20
+ 224,
21
+ 224
22
+ ]
23
+ }
runs/Apr20_09-46-25_c768f2f23310/events.out.tfevents.1713606387.c768f2f23310.228.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1bf0fd06504c687133b989429f35b0d5541543d69ec0971e7a73e292ca1242e1
3
+ size 4685
runs/Apr20_09-47-33_c768f2f23310/events.out.tfevents.1713606455.c768f2f23310.228.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:764464100f34488d0846422b61cabbeb65729c566fe180f24b4f0f16ffbb9d11
3
+ size 4734
runs/Apr20_09-48-37_c768f2f23310/events.out.tfevents.1713606519.c768f2f23310.228.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:762bec7312ea3e8bf078e2faeaaab5b96553baf1ab651a644003a7b97611c4d3
3
+ size 4733
runs/Apr20_09-53-52_c768f2f23310/events.out.tfevents.1713606833.c768f2f23310.228.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea9098b9ec7d10c3990a42faec85c69b35bdd10132debb75f4707a4ec7363153
3
+ size 4734
runs/Apr20_09-54-21_c768f2f23310/events.out.tfevents.1713606863.c768f2f23310.228.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d45609c8be6f1a8ad1f6157d18944150eb884f6c999e4e32bb0f24b8498e8628
3
+ size 16808
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74b8c200f5764b82c5e7eae56355274a3af515d84f07b543a6e6bce65aca1f26
3
+ size 4856