Hugging Face's logo

TirathP
/
finetuned_model

Image Classification
Transformers
PyTorch
vit
Generated from Trainer
Model card Files
xet
Community
1
TirathP commited on
Commit
dd1e850
·
verified ·
1 Parent(s): f07cbee

Model save

Browse files
Files changed (7) hide show
  1. all_results.json +10 -10
  2. config.json +26 -22
  3. eval_results.json +6 -6
  4. pytorch_model.bin +2 -2
  5. train_results.json +4 -4
  6. trainer_state.json +4 -4
  7. training_args.bin +1 -1
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 30.0,
3
- "eval_accuracy": 0.14285714285714285,
4
- "eval_f1": 0.022727272727272728,
5
- "eval_loss": 2.3439788818359375,
6
- "eval_runtime": 6.9153,
7
- "eval_samples_per_second": 2.024,
8
- "eval_steps_per_second": 0.289,
9
- "train_loss": 3.281195576985677,
10
- "train_runtime": 379.7434,
11
- "train_samples_per_second": 1.106,
12
- "train_steps_per_second": 0.079
13
  }
 
1
  {
2
  "epoch": 30.0,
3
+ "eval_accuracy": 0.42857142857142855,
4
+ "eval_f1": 0.2852813852813853,
5
+ "eval_loss": 0.9329891800880432,
6
+ "eval_runtime": 9.6641,
7
+ "eval_samples_per_second": 1.449,
8
+ "eval_steps_per_second": 0.207,
9
+ "train_loss": 1.0424017588297525,
10
+ "train_runtime": 383.8523,
11
+ "train_samples_per_second": 1.094,
12
+ "train_steps_per_second": 0.078
13
  }
config.json CHANGED
@@ -9,33 +9,37 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "boat",
13
- "1": "children",
14
- "10": "water",
15
- "2": "dogs",
16
- "3": "fireman",
17
- "4": "firetruck",
18
- "5": "mountains",
19
- "6": "people",
20
- "7": "river",
21
- "8": "snow",
22
- "9": "stairs"
 
 
23
  },
24
  "image_size": 224,
25
  "initializer_range": 0.02,
26
  "intermediate_size": 3072,
27
  "label2id": {
28
- "boat": "0",
29
- "children": "1",
30
- "dogs": "2",
31
- "fireman": "3",
32
- "firetruck": "4",
33
- "mountains": "5",
34
- "people": "6",
35
- "river": "7",
36
- "snow": "8",
37
- "stairs": "9",
38
- "water": "10"
 
 
39
  },
40
  "layer_norm_eps": 1e-12,
41
  "model_type": "vit",
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "aeroplane",
13
+ "1": "boat",
14
+ "10": "stairs",
15
+ "11": "tree",
16
+ "12": "water",
17
+ "2": "children",
18
+ "3": "dogs",
19
+ "4": "fireman",
20
+ "5": "firetruck",
21
+ "6": "mountains",
22
+ "7": "people",
23
+ "8": "river",
24
+ "9": "snow"
25
  },
26
  "image_size": 224,
27
  "initializer_range": 0.02,
28
  "intermediate_size": 3072,
29
  "label2id": {
30
+ "aeroplane": "0",
31
+ "boat": "1",
32
+ "children": "2",
33
+ "dogs": "3",
34
+ "fireman": "4",
35
+ "firetruck": "5",
36
+ "mountains": "6",
37
+ "people": "7",
38
+ "river": "8",
39
+ "snow": "9",
40
+ "stairs": "10",
41
+ "tree": "11",
42
+ "water": "12"
43
  },
44
  "layer_norm_eps": 1e-12,
45
  "model_type": "vit",
eval_results.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "epoch": 30.0,
3
- "eval_accuracy": 0.14285714285714285,
4
- "eval_f1": 0.022727272727272728,
5
- "eval_loss": 2.3439788818359375,
6
- "eval_runtime": 6.9153,
7
- "eval_samples_per_second": 2.024,
8
- "eval_steps_per_second": 0.289
9
  }
 
1
  {
2
  "epoch": 30.0,
3
+ "eval_accuracy": 0.42857142857142855,
4
+ "eval_f1": 0.2852813852813853,
5
+ "eval_loss": 0.9329891800880432,
6
+ "eval_runtime": 9.6641,
7
+ "eval_samples_per_second": 1.449,
8
+ "eval_steps_per_second": 0.207
9
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:24e4586af2d776dda4f6887d2efab872ad90bcdf5bd1f92ca2adbc7b2a2de230
3
- size 343293613
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe96c949e5ad21cb5a030a5afd86937f73a33e1b3fcf8a0bfe6731d9b92b2674
3
+ size 343299757
train_results.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "epoch": 30.0,
3
- "train_loss": 3.281195576985677,
4
- "train_runtime": 379.7434,
5
- "train_samples_per_second": 1.106,
6
- "train_steps_per_second": 0.079
7
  }
 
1
  {
2
  "epoch": 30.0,
3
+ "train_loss": 1.0424017588297525,
4
+ "train_runtime": 383.8523,
5
+ "train_samples_per_second": 1.094,
6
+ "train_steps_per_second": 0.078
7
  }
trainer_state.json CHANGED
@@ -12,10 +12,10 @@
12
  "epoch": 30.0,
13
  "step": 30,
14
  "total_flos": 3.25492609849344e+16,
15
- "train_loss": 3.281195576985677,
16
- "train_runtime": 379.7434,
17
- "train_samples_per_second": 1.106,
18
- "train_steps_per_second": 0.079
19
  }
20
  ],
21
  "logging_steps": 1000,
 
12
  "epoch": 30.0,
13
  "step": 30,
14
  "total_flos": 3.25492609849344e+16,
15
+ "train_loss": 1.0424017588297525,
16
+ "train_runtime": 383.8523,
17
+ "train_samples_per_second": 1.094,
18
+ "train_steps_per_second": 0.078
19
  }
20
  ],
21
  "logging_steps": 1000,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f4e76ab48f6044883bc0d7303bbc1262a38464a62f519def1c6306f002a2f850
3
  size 4027
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0cf2287a18cc72d9e06aacf3d85a4179b9a3cfbee45d74f8950c4b77d87c89af
3
  size 4027