Pamreth commited on
Commit
9117cbd
·
verified ·
1 Parent(s): 1b8e80b

🍻 cheers

Browse files
README.md ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/vit-base-patch16-224-in21k
5
+ tags:
6
+ - image-classification
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: vit-ena24
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
+ # vit-ena24
17
+
18
+ This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the beans dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - eval_loss: 3.0866
21
+ - eval_model_preparation_time: 0.0047
22
+ - eval_accuracy: 0.0473
23
+ - eval_runtime: 917.1232
24
+ - eval_samples_per_second: 1.428
25
+ - eval_steps_per_second: 0.179
26
+ - step: 0
27
+
28
+ ## Model description
29
+
30
+ More information needed
31
+
32
+ ## Intended uses & limitations
33
+
34
+ More information needed
35
+
36
+ ## Training and evaluation data
37
+
38
+ More information needed
39
+
40
+ ## Training procedure
41
+
42
+ ### Training hyperparameters
43
+
44
+ The following hyperparameters were used during training:
45
+ - learning_rate: 0.0002
46
+ - train_batch_size: 8
47
+ - eval_batch_size: 8
48
+ - seed: 42
49
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
50
+ - lr_scheduler_type: linear
51
+ - num_epochs: 2
52
+ - mixed_precision_training: Native AMP
53
+
54
+ ### Framework versions
55
+
56
+ - Transformers 4.51.1
57
+ - Pytorch 2.6.0+cu124
58
+ - Datasets 3.5.0
59
+ - Tokenizers 0.21.1
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_accuracy": 0.04732824427480916,
3
+ "eval_loss": 3.08662486076355,
4
+ "eval_model_preparation_time": 0.0047,
5
+ "eval_runtime": 917.1232,
6
+ "eval_samples_per_second": 1.428,
7
+ "eval_steps_per_second": 0.179
8
+ }
config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTForImageClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "encoder_stride": 16,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.0,
9
+ "hidden_size": 768,
10
+ "id2label": {
11
+ "0": "American Black Bear",
12
+ "1": "American Crow",
13
+ "10": "Eastern Fox Squirrel",
14
+ "11": "Eastern Gray Squirrel",
15
+ "12": "Grey Fox",
16
+ "13": "Horse",
17
+ "14": "Northern Raccoon",
18
+ "15": "Red Fox",
19
+ "16": "Striped Skunk",
20
+ "17": "Vehicle",
21
+ "18": "Virginia Opossum",
22
+ "19": "White_Tailed_Deer",
23
+ "2": "Bird",
24
+ "20": "Wild Turkey",
25
+ "21": "Woodchuck",
26
+ "3": "Bobcat",
27
+ "4": "Chicken",
28
+ "5": "Coyote",
29
+ "6": "Dog",
30
+ "7": "Domestic Cat",
31
+ "8": "Eastern Chipmunk",
32
+ "9": "Eastern Cottontail"
33
+ },
34
+ "image_size": 224,
35
+ "initializer_range": 0.02,
36
+ "intermediate_size": 3072,
37
+ "label2id": {
38
+ "American Black Bear": "0",
39
+ "American Crow": "1",
40
+ "Bird": "2",
41
+ "Bobcat": "3",
42
+ "Chicken": "4",
43
+ "Coyote": "5",
44
+ "Dog": "6",
45
+ "Domestic Cat": "7",
46
+ "Eastern Chipmunk": "8",
47
+ "Eastern Cottontail": "9",
48
+ "Eastern Fox Squirrel": "10",
49
+ "Eastern Gray Squirrel": "11",
50
+ "Grey Fox": "12",
51
+ "Horse": "13",
52
+ "Northern Raccoon": "14",
53
+ "Red Fox": "15",
54
+ "Striped Skunk": "16",
55
+ "Vehicle": "17",
56
+ "Virginia Opossum": "18",
57
+ "White_Tailed_Deer": "19",
58
+ "Wild Turkey": "20",
59
+ "Woodchuck": "21"
60
+ },
61
+ "layer_norm_eps": 1e-12,
62
+ "model_type": "vit",
63
+ "num_attention_heads": 12,
64
+ "num_channels": 3,
65
+ "num_hidden_layers": 12,
66
+ "patch_size": 16,
67
+ "pooler_act": "tanh",
68
+ "pooler_output_size": 768,
69
+ "problem_type": "single_label_classification",
70
+ "qkv_bias": true,
71
+ "torch_dtype": "float32",
72
+ "transformers_version": "4.51.1"
73
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_accuracy": 0.04732824427480916,
3
+ "eval_loss": 3.08662486076355,
4
+ "eval_model_preparation_time": 0.0047,
5
+ "eval_runtime": 917.1232,
6
+ "eval_samples_per_second": 1.428,
7
+ "eval_steps_per_second": 0.179
8
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df275aaaa1509717b68f5cb779f769300ce9faf581adadb7179e8589b46fac4f
3
+ size 343285496
preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "ViTImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 224,
21
+ "width": 224
22
+ }
23
+ }
runs/Apr14_17-08-40_f3935f1f5266/events.out.tfevents.1744651561.f3935f1f5266.1199.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98d528eabcb3c4e870a8b3c0a599ab606f604e88eab556c5887f87139535e4c1
3
+ size 409
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d78ea8cb4472688a53d74cae5dc602594965a09a43bf10b39982546ef1e0e5
3
+ size 5304