Hugging Face's logo

dingusagar
/
vit-base-avengers-v2

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results (legacy)
Model card Files
xet
 Metrics Community
dingusagar commited on
Commit
87c5915
·
verified ·
1 Parent(s): a825013

End of training

Browse files
Files changed (6) hide show
  1. README.md +19 -1
  2. all_results.json +10 -10
  3. eval_results.json +5 -5
  4. runs/Jan27_11-04-29_93638ce2fa1a/events.out.tfevents.1706353569.93638ce2fa1a.145.1 +3 -0
  5. train_results.json +5 -5
  6. trainer_state.json +14 -14
README.md CHANGED
@@ -5,9 +5,24 @@ tags:
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
 
 
8
  model-index:
9
  - name: vit-base-avengers-v2
10
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ---
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,6 +31,9 @@ should probably proofread and complete it, then remove this comment. -->
16
  # vit-base-avengers-v2
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 imagefolder dataset.
 
 
 
19
 
20
  ## Model description
21
 
 
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
8
+ metrics:
9
+ - accuracy
10
  model-index:
11
  - name: vit-base-avengers-v2
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: imagefolder
18
+ type: imagefolder
19
+ config: default
20
+ split: test
21
+ args: default
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.9125
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
  # vit-base-avengers-v2
32
 
33
  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 imagefolder dataset.
34
+ It achieves the following results on the evaluation set:
35
+ - Loss: 0.2542
36
+ - Accuracy: 0.9125
37
 
38
  ## Model description
39
 
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 4.0,
3
- "eval_accuracy": 0.8974358974358975,
4
- "eval_loss": 0.2478559911251068,
5
- "eval_runtime": 3.4307,
6
- "eval_samples_per_second": 22.736,
7
- "eval_steps_per_second": 2.915,
8
- "total_flos": 9.516186902760653e+16,
9
- "train_loss": 0.3570560812950134,
10
- "train_runtime": 67.9287,
11
- "train_samples_per_second": 18.078,
12
- "train_steps_per_second": 1.178
13
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.9125,
4
+ "eval_loss": 0.2541953921318054,
5
+ "eval_runtime": 2.9203,
6
+ "eval_samples_per_second": 27.394,
7
+ "eval_steps_per_second": 3.424,
8
+ "total_flos": 9.547184254235443e+16,
9
+ "train_loss": 0.37343859672546387,
10
+ "train_runtime": 76.2656,
11
+ "train_samples_per_second": 16.154,
12
+ "train_steps_per_second": 1.049
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 4.0,
3
- "eval_accuracy": 0.8974358974358975,
4
- "eval_loss": 0.2478559911251068,
5
- "eval_runtime": 3.4307,
6
- "eval_samples_per_second": 22.736,
7
- "eval_steps_per_second": 2.915
8
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.9125,
4
+ "eval_loss": 0.2541953921318054,
5
+ "eval_runtime": 2.9203,
6
+ "eval_samples_per_second": 27.394,
7
+ "eval_steps_per_second": 3.424
8
  }
runs/Jan27_11-04-29_93638ce2fa1a/events.out.tfevents.1706353569.93638ce2fa1a.145.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f033a87fb35a5c784db87ec4572f77a72cf88d2b1af9cd62c193db2d2571325
3
+ size 405
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 4.0,
3
- "total_flos": 9.516186902760653e+16,
4
- "train_loss": 0.3570560812950134,
5
- "train_runtime": 67.9287,
6
- "train_samples_per_second": 18.078,
7
- "train_steps_per_second": 1.178
8
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "total_flos": 9.547184254235443e+16,
4
+ "train_loss": 0.37343859672546387,
5
+ "train_runtime": 76.2656,
6
+ "train_samples_per_second": 16.154,
7
+ "train_steps_per_second": 1.049
8
  }
trainer_state.json CHANGED
@@ -11,59 +11,59 @@
11
  {
12
  "epoch": 0.5,
13
  "learning_rate": 0.000175,
14
- "loss": 1.192,
15
  "step": 10
16
  },
17
  {
18
  "epoch": 1.0,
19
  "learning_rate": 0.00015000000000000001,
20
- "loss": 0.8314,
21
  "step": 20
22
  },
23
  {
24
  "epoch": 1.5,
25
  "learning_rate": 0.000125,
26
- "loss": 0.3303,
27
  "step": 30
28
  },
29
  {
30
  "epoch": 2.0,
31
  "learning_rate": 0.0001,
32
- "loss": 0.199,
33
  "step": 40
34
  },
35
  {
36
  "epoch": 2.5,
37
  "learning_rate": 7.500000000000001e-05,
38
- "loss": 0.1008,
39
  "step": 50
40
  },
41
  {
42
  "epoch": 3.0,
43
  "learning_rate": 5e-05,
44
- "loss": 0.0759,
45
  "step": 60
46
  },
47
  {
48
  "epoch": 3.5,
49
  "learning_rate": 2.5e-05,
50
- "loss": 0.0652,
51
  "step": 70
52
  },
53
  {
54
  "epoch": 4.0,
55
  "learning_rate": 0.0,
56
- "loss": 0.0619,
57
  "step": 80
58
  },
59
  {
60
  "epoch": 4.0,
61
  "step": 80,
62
- "total_flos": 9.516186902760653e+16,
63
- "train_loss": 0.3570560812950134,
64
- "train_runtime": 67.9287,
65
- "train_samples_per_second": 18.078,
66
- "train_steps_per_second": 1.178
67
  }
68
  ],
69
  "logging_steps": 10,
@@ -71,7 +71,7 @@
71
  "num_input_tokens_seen": 0,
72
  "num_train_epochs": 4,
73
  "save_steps": 100,
74
- "total_flos": 9.516186902760653e+16,
75
  "train_batch_size": 16,
76
  "trial_name": null,
77
  "trial_params": null
 
11
  {
12
  "epoch": 0.5,
13
  "learning_rate": 0.000175,
14
+ "loss": 1.2435,
15
  "step": 10
16
  },
17
  {
18
  "epoch": 1.0,
19
  "learning_rate": 0.00015000000000000001,
20
+ "loss": 0.7842,
21
  "step": 20
22
  },
23
  {
24
  "epoch": 1.5,
25
  "learning_rate": 0.000125,
26
+ "loss": 0.3928,
27
  "step": 30
28
  },
29
  {
30
  "epoch": 2.0,
31
  "learning_rate": 0.0001,
32
+ "loss": 0.2174,
33
  "step": 40
34
  },
35
  {
36
  "epoch": 2.5,
37
  "learning_rate": 7.500000000000001e-05,
38
+ "loss": 0.1137,
39
  "step": 50
40
  },
41
  {
42
  "epoch": 3.0,
43
  "learning_rate": 5e-05,
44
+ "loss": 0.0931,
45
  "step": 60
46
  },
47
  {
48
  "epoch": 3.5,
49
  "learning_rate": 2.5e-05,
50
+ "loss": 0.0725,
51
  "step": 70
52
  },
53
  {
54
  "epoch": 4.0,
55
  "learning_rate": 0.0,
56
+ "loss": 0.0704,
57
  "step": 80
58
  },
59
  {
60
  "epoch": 4.0,
61
  "step": 80,
62
+ "total_flos": 9.547184254235443e+16,
63
+ "train_loss": 0.37343859672546387,
64
+ "train_runtime": 76.2656,
65
+ "train_samples_per_second": 16.154,
66
+ "train_steps_per_second": 1.049
67
  }
68
  ],
69
  "logging_steps": 10,
 
71
  "num_input_tokens_seen": 0,
72
  "num_train_epochs": 4,
73
  "save_steps": 100,
74
+ "total_flos": 9.547184254235443e+16,
75
  "train_batch_size": 16,
76
  "trial_name": null,
77
  "trial_params": null