Hugging Face's logo

Kushagra07
/
autotrain-vit-base-patch16-224

Image Classification
Transformers
TensorBoard
Safetensors
vit
Trained with AutoTrain
Inference Endpoints
Model card Files Metrics Community
Kushagra07 commited on
Commit
77f3b63
•
1 Parent(s): d9a8eea

Upload folder using huggingface_hub

Browse files
Files changed (15) hide show
  1. README.md +42 -0
  2. checkpoint-14620/config.json +63 -0
  3. checkpoint-14620/model.safetensors +3 -0
  4. checkpoint-14620/optimizer.pt +3 -0
  5. checkpoint-14620/rng_state.pth +3 -0
  6. checkpoint-14620/scheduler.pt +3 -0
  7. checkpoint-14620/trainer_state.json +0 -0
  8. checkpoint-14620/training_args.bin +3 -0
  9. config.json +63 -0
  10. model.safetensors +3 -0
  11. preprocessor_config.json +36 -0
  12. runs/Apr30_07-03-12_d7c8b0ba738f/events.out.tfevents.1714460593.d7c8b0ba738f.17115.0 +2 -2
  13. runs/Apr30_07-03-12_d7c8b0ba738f/events.out.tfevents.1714464461.d7c8b0ba738f.17115.1 +3 -0
  14. training_args.bin +3 -0
  15. training_params.json +27 -0
README.md ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - image-classification
6
+ widget:
7
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
8
+ example_title: Tiger
9
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
10
+ example_title: Teapot
11
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
12
+ example_title: Palace
13
+ datasets:
14
+ - autotrain-vit-base-patch16-224/autotrain-data
15
+ ---
16
+
17
+ # Model Trained Using AutoTrain
18
+
19
+ - Problem type: Image Classification
20
+
21
+ ## Validation Metrics
22
+ loss: 0.223019078373909
23
+
24
+ f1_macro: 0.8656901233453752
25
+
26
+ f1_micro: 0.9238648002731308
27
+
28
+ f1_weighted: 0.9239248911195606
29
+
30
+ precision_macro: 0.928029530990028
31
+
32
+ precision_micro: 0.9238648002731308
33
+
34
+ precision_weighted: 0.9287629201629745
35
+
36
+ recall_macro: 0.834713663096659
37
+
38
+ recall_micro: 0.9238648002731308
39
+
40
+ recall_weighted: 0.9238648002731308
41
+
42
+ accuracy: 0.9238648002731308
checkpoint-14620/config.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
+ "_num_labels": 17,
4
+ "architectures": [
5
+ "ViTForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "encoder_stride": 16,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "AadhaarBack",
14
+ "1": "AadhaarBackMasked",
15
+ "2": "AadhaarFront",
16
+ "3": "AadhaarFrontMasked",
17
+ "4": "AadhaarRegular",
18
+ "5": "AadhaarRegularMasked",
19
+ "6": "AadhaarSelfGenerated",
20
+ "7": "DrivingLicenseNew",
21
+ "8": "DrivingLicenseOld",
22
+ "9": "MultipleOVDs",
23
+ "10": "PanCard",
24
+ "11": "PassportFirst",
25
+ "12": "PassportLast",
26
+ "13": "PassportRegular",
27
+ "14": "VoterCardBack",
28
+ "15": "VoterCardFront",
29
+ "16": "VoterCardRegular"
30
+ },
31
+ "image_size": 224,
32
+ "initializer_range": 0.02,
33
+ "intermediate_size": 3072,
34
+ "label2id": {
35
+ "AadhaarBack": 0,
36
+ "AadhaarBackMasked": 1,
37
+ "AadhaarFront": 2,
38
+ "AadhaarFrontMasked": 3,
39
+ "AadhaarRegular": 4,
40
+ "AadhaarRegularMasked": 5,
41
+ "AadhaarSelfGenerated": 6,
42
+ "DrivingLicenseNew": 7,
43
+ "DrivingLicenseOld": 8,
44
+ "MultipleOVDs": 9,
45
+ "PanCard": 10,
46
+ "PassportFirst": 11,
47
+ "PassportLast": 12,
48
+ "PassportRegular": 13,
49
+ "VoterCardBack": 14,
50
+ "VoterCardFront": 15,
51
+ "VoterCardRegular": 16
52
+ },
53
+ "layer_norm_eps": 1e-12,
54
+ "model_type": "vit",
55
+ "num_attention_heads": 12,
56
+ "num_channels": 3,
57
+ "num_hidden_layers": 12,
58
+ "patch_size": 16,
59
+ "problem_type": "single_label_classification",
60
+ "qkv_bias": true,
61
+ "torch_dtype": "float32",
62
+ "transformers_version": "4.40.1"
63
+ }
checkpoint-14620/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40e9f886fd3605ce12b7d3581f8a1c4e440a1340a5bd46701d8f4cd9973f3987
3
+ size 343270116
checkpoint-14620/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:100f372b6646bc156f3c354e49854be3df43c98aa1f61129de4a0fb64020940c
3
+ size 686661178
checkpoint-14620/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81159f2ad62765014420fbf9c18708fd2f64751956556e519fc9c9531e06491b
3
+ size 14244
checkpoint-14620/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7074a2f9eeff361f67855c4c4665de114e54ca1649449512dfb23ca0f2344485
3
+ size 1064
checkpoint-14620/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-14620/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:156d8ed9dda340d2acf4fc0f170d9fc77892f03dfe903acca95533d54128b51a
3
+ size 5048
config.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
+ "_num_labels": 17,
4
+ "architectures": [
5
+ "ViTForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "encoder_stride": 16,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "AadhaarBack",
14
+ "1": "AadhaarBackMasked",
15
+ "2": "AadhaarFront",
16
+ "3": "AadhaarFrontMasked",
17
+ "4": "AadhaarRegular",
18
+ "5": "AadhaarRegularMasked",
19
+ "6": "AadhaarSelfGenerated",
20
+ "7": "DrivingLicenseNew",
21
+ "8": "DrivingLicenseOld",
22
+ "9": "MultipleOVDs",
23
+ "10": "PanCard",
24
+ "11": "PassportFirst",
25
+ "12": "PassportLast",
26
+ "13": "PassportRegular",
27
+ "14": "VoterCardBack",
28
+ "15": "VoterCardFront",
29
+ "16": "VoterCardRegular"
30
+ },
31
+ "image_size": 224,
32
+ "initializer_range": 0.02,
33
+ "intermediate_size": 3072,
34
+ "label2id": {
35
+ "AadhaarBack": 0,
36
+ "AadhaarBackMasked": 1,
37
+ "AadhaarFront": 2,
38
+ "AadhaarFrontMasked": 3,
39
+ "AadhaarRegular": 4,
40
+ "AadhaarRegularMasked": 5,
41
+ "AadhaarSelfGenerated": 6,
42
+ "DrivingLicenseNew": 7,
43
+ "DrivingLicenseOld": 8,
44
+ "MultipleOVDs": 9,
45
+ "PanCard": 10,
46
+ "PassportFirst": 11,
47
+ "PassportLast": 12,
48
+ "PassportRegular": 13,
49
+ "VoterCardBack": 14,
50
+ "VoterCardFront": 15,
51
+ "VoterCardRegular": 16
52
+ },
53
+ "layer_norm_eps": 1e-12,
54
+ "model_type": "vit",
55
+ "num_attention_heads": 12,
56
+ "num_channels": 3,
57
+ "num_hidden_layers": 12,
58
+ "patch_size": 16,
59
+ "problem_type": "single_label_classification",
60
+ "qkv_bias": true,
61
+ "torch_dtype": "float32",
62
+ "transformers_version": "4.40.1"
63
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40e9f886fd3605ce12b7d3581f8a1c4e440a1340a5bd46701d8f4cd9973f3987
3
+ size 343270116
preprocessor_config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "do_resize",
5
+ "size",
6
+ "resample",
7
+ "do_rescale",
8
+ "rescale_factor",
9
+ "do_normalize",
10
+ "image_mean",
11
+ "image_std",
12
+ "return_tensors",
13
+ "data_format",
14
+ "input_data_format"
15
+ ],
16
+ "do_normalize": true,
17
+ "do_rescale": true,
18
+ "do_resize": true,
19
+ "image_mean": [
20
+ 0.5,
21
+ 0.5,
22
+ 0.5
23
+ ],
24
+ "image_processor_type": "ViTImageProcessor",
25
+ "image_std": [
26
+ 0.5,
27
+ 0.5,
28
+ 0.5
29
+ ],
30
+ "resample": 2,
31
+ "rescale_factor": 0.00392156862745098,
32
+ "size": {
33
+ "height": 224,
34
+ "width": 224
35
+ }
36
+ }
runs/Apr30_07-03-12_d7c8b0ba738f/events.out.tfevents.1714460593.d7c8b0ba738f.17115.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:271c4f9dd68a7b695f13a1745177d45cffd97a0d5cc6f69f8e4825f6c2f2fa72
3
- size 132058
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf72bcc28ae834e88a9baf1a658bfa9e90d24042f2dd686dbf8c55e89c94deea
3
+ size 137465
runs/Apr30_07-03-12_d7c8b0ba738f/events.out.tfevents.1714464461.d7c8b0ba738f.17115.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdbf9188290bc7022009fed15ed6a9ab117fea3741ac24a55d6e8294d61ff692
3
+ size 921
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:156d8ed9dda340d2acf4fc0f170d9fc77892f03dfe903acca95533d54128b51a
3
+ size 5048
training_params.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "autotrain-vit-base-patch16-224/autotrain-data",
3
+ "model": "google/vit-base-patch16-224",
4
+ "username": "Kushagra07",
5
+ "lr": 5e-05,
6
+ "epochs": 10,
7
+ "batch_size": 8,
8
+ "warmup_ratio": 0.1,
9
+ "gradient_accumulation": 1,
10
+ "optimizer": "adamw_torch",
11
+ "scheduler": "linear",
12
+ "weight_decay": 0.0,
13
+ "max_grad_norm": 1.0,
14
+ "seed": 42,
15
+ "train_split": "train",
16
+ "valid_split": "validation",
17
+ "logging_steps": -1,
18
+ "project_name": "autotrain-vit-base-patch16-224",
19
+ "auto_find_batch_size": true,
20
+ "mixed_precision": "fp16",
21
+ "save_total_limit": 1,
22
+ "push_to_hub": true,
23
+ "evaluation_strategy": "epoch",
24
+ "image_column": "autotrain_image",
25
+ "target_column": "autotrain_label",
26
+ "log": "tensorboard"
27
+ }