Hugging Face's logo

sooks
/
idbwbase

Image Classification
Transformers
TensorBoard
Safetensors
convnext
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
sooks commited on
Commit
fe0342c
·
verified ·
1 Parent(s): 8fafb36

Model save

Browse files
Files changed (9) hide show
  1. README.md +70 -0
  2. config.json +52 -0
  3. model.safetensors +3 -0
  4. preprocessor_config.json +23 -0
  5. runs/Apr20_09-46-50_49f104673c10/events.out.tfevents.1713606413.49f104673c10.914.0 +3 -0
  6. runs/Apr20_09-47-45_49f104673c10/events.out.tfevents.1713606467.49f104673c10.914.1 +3 -0
  7. runs/Apr20_09-49-02_49f104673c10/events.out.tfevents.1713606544.49f104673c10.914.2 +3 -0
  8. runs/Apr20_09-53-02_49f104673c10/events.out.tfevents.1713606784.49f104673c10.914.3 +3 -0
  9. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: facebook/convnext-base-224-22k
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ model-index:
9
+ - name: idbwbase
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
+ # idbwbase
17
+
18
+ This model is a fine-tuned version of [facebook/convnext-base-224-22k](https://huggingface.co/facebook/convnext-base-224-22k) on an unknown dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.1001
21
+ - Accuracy: 0.9761
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: 64
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.373 | 1.0 | 4709 | 0.3002 | 0.8705 |
54
+ | 0.3244 | 2.0 | 9418 | 0.2262 | 0.9044 |
55
+ | 0.2801 | 3.0 | 14127 | 0.1987 | 0.9196 |
56
+ | 0.2366 | 4.0 | 18836 | 0.1788 | 0.9345 |
57
+ | 0.2051 | 5.0 | 23545 | 0.1463 | 0.9484 |
58
+ | 0.1764 | 6.0 | 28254 | 0.1202 | 0.9593 |
59
+ | 0.1595 | 7.0 | 32963 | 0.1243 | 0.9655 |
60
+ | 0.1359 | 8.0 | 37672 | 0.1188 | 0.9659 |
61
+ | 0.1231 | 9.0 | 42381 | 0.0978 | 0.9741 |
62
+ | 0.1162 | 10.0 | 47090 | 0.1001 | 0.9761 |
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-base-224-22k",
3
+ "architectures": [
4
+ "ConvNextForImageClassification"
5
+ ],
6
+ "depths": [
7
+ 3,
8
+ 3,
9
+ 27,
10
+ 3
11
+ ],
12
+ "drop_path_rate": 0.0,
13
+ "hidden_act": "gelu",
14
+ "hidden_sizes": [
15
+ 128,
16
+ 256,
17
+ 512,
18
+ 1024
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:0bc803e7f13d8b8db8a5efd2c26cae74afacaf5896b0b7b3f7aa58bd082c7a92
3
+ size 350315208
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-50_49f104673c10/events.out.tfevents.1713606413.49f104673c10.914.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4bfdea8a0eb9a8dcc77a3482ff5a8a7ce0018fc4b0faf8eb5bc1145b139fff4d
3
+ size 4692
runs/Apr20_09-47-45_49f104673c10/events.out.tfevents.1713606467.49f104673c10.914.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9c26bc5759c91013d0e0f7bfa91f34eeb3c60347849b0ad44cbd1dc78d64233
3
+ size 4692
runs/Apr20_09-49-02_49f104673c10/events.out.tfevents.1713606544.49f104673c10.914.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:505bd22b3fed79db9cb91b6391b89184aba5258f2eace31cb050c1c966dfbeba
3
+ size 4691
runs/Apr20_09-53-02_49f104673c10/events.out.tfevents.1713606784.49f104673c10.914.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9b60e8698c1c9375655a6ee795e248bea70103653a56cf08253e849d740c3c9
3
+ size 28454
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef635a4a97634b4feedaa4e5cd63c754f17e03e39e81c5a4fd91c3f53428f6ac
3
+ size 4856