Hugging Face's logo

rhlshah
/
cv-module

Image Classification
Transformers
PyTorch
vit
Model card Files
xet
Community
1
rhlshah commited on
Commit
57cdd05
·
1 Parent(s): 3f6e73d

Upload 7 files

Browse files
Files changed (7) hide show
  1. all_results.json +8 -0
  2. config.json +32 -0
  3. preprocessor_config.json +22 -0
  4. pytorch_model.bin +3 -0
  5. train_results.json +8 -0
  6. trainer_state.json +103 -0
  7. training_args.bin +3 -0
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "total_flos": 1.2057753584030515e+17,
4
+ "train_loss": 0.030779192969202996,
5
+ "train_runtime": 74.2102,
6
+ "train_samples_per_second": 20.967,
7
+ "train_steps_per_second": 1.348
8
+ }
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "Blight",
13
+ "1": "Healthy"
14
+ },
15
+ "image_size": 224,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
+ "label2id": {
19
+ "Blight": "0",
20
+ "Healthy": "1"
21
+ },
22
+ "layer_norm_eps": 1e-12,
23
+ "model_type": "vit",
24
+ "num_attention_heads": 12,
25
+ "num_channels": 3,
26
+ "num_hidden_layers": 12,
27
+ "patch_size": 16,
28
+ "problem_type": "single_label_classification",
29
+ "qkv_bias": true,
30
+ "torch_dtype": "float32",
31
+ "transformers_version": "4.27.1"
32
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTFeatureExtractor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66121eedfca02a410d5256d2fa7852455fb5df232a65de6625504b2813c89442
3
+ size 343268717
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "total_flos": 1.2057753584030515e+17,
4
+ "train_loss": 0.030779192969202996,
5
+ "train_runtime": 74.2102,
6
+ "train_samples_per_second": 20.967,
7
+ "train_steps_per_second": 1.348
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.0044614835642278194,
3
+ "best_model_checkpoint": "./vit-base-beans/checkpoint-80",
4
+ "epoch": 4.0,
5
+ "global_step": 100,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.4,
12
+ "learning_rate": 0.00018,
13
+ "loss": 0.244,
14
+ "step": 10
15
+ },
16
+ {
17
+ "epoch": 0.8,
18
+ "learning_rate": 0.00016,
19
+ "loss": 0.019,
20
+ "step": 20
21
+ },
22
+ {
23
+ "epoch": 1.2,
24
+ "learning_rate": 0.00014,
25
+ "loss": 0.0092,
26
+ "step": 30
27
+ },
28
+ {
29
+ "epoch": 1.6,
30
+ "learning_rate": 0.00012,
31
+ "loss": 0.0068,
32
+ "step": 40
33
+ },
34
+ {
35
+ "epoch": 1.6,
36
+ "eval_accuracy": 1.0,
37
+ "eval_loss": 0.006160833407193422,
38
+ "eval_runtime": 1.4649,
39
+ "eval_samples_per_second": 75.773,
40
+ "eval_steps_per_second": 9.557,
41
+ "step": 40
42
+ },
43
+ {
44
+ "epoch": 2.0,
45
+ "learning_rate": 0.0001,
46
+ "loss": 0.0058,
47
+ "step": 50
48
+ },
49
+ {
50
+ "epoch": 2.4,
51
+ "learning_rate": 8e-05,
52
+ "loss": 0.0051,
53
+ "step": 60
54
+ },
55
+ {
56
+ "epoch": 2.8,
57
+ "learning_rate": 6e-05,
58
+ "loss": 0.0048,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 3.2,
63
+ "learning_rate": 4e-05,
64
+ "loss": 0.0045,
65
+ "step": 80
66
+ },
67
+ {
68
+ "epoch": 3.2,
69
+ "eval_accuracy": 1.0,
70
+ "eval_loss": 0.0044614835642278194,
71
+ "eval_runtime": 1.5025,
72
+ "eval_samples_per_second": 73.876,
73
+ "eval_steps_per_second": 9.318,
74
+ "step": 80
75
+ },
76
+ {
77
+ "epoch": 3.6,
78
+ "learning_rate": 2e-05,
79
+ "loss": 0.0044,
80
+ "step": 90
81
+ },
82
+ {
83
+ "epoch": 4.0,
84
+ "learning_rate": 0.0,
85
+ "loss": 0.0043,
86
+ "step": 100
87
+ },
88
+ {
89
+ "epoch": 4.0,
90
+ "step": 100,
91
+ "total_flos": 1.2057753584030515e+17,
92
+ "train_loss": 0.030779192969202996,
93
+ "train_runtime": 74.2102,
94
+ "train_samples_per_second": 20.967,
95
+ "train_steps_per_second": 1.348
96
+ }
97
+ ],
98
+ "max_steps": 100,
99
+ "num_train_epochs": 4,
100
+ "total_flos": 1.2057753584030515e+17,
101
+ "trial_name": null,
102
+ "trial_params": null
103
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82c09b202fe06f84fd1cb68dbd5ef13d6178af156d5d7852e84644fee8508e6c
3
+ size 3515