ChrisGuarino
commited on
Commit
•
641b556
1
Parent(s):
e01cac9
ChrisGuarino/yummy_model
Browse files- README.md +6 -6
- all_results.json +4 -4
- model.safetensors +1 -1
- train_results.json +4 -4
- trainer_state.json +26 -26
- training_args.bin +1 -1
README.md
CHANGED
@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
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 an unknown dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
-
- Loss: 0.
|
21 |
- Accuracy: 1.0
|
22 |
|
23 |
## Model description
|
@@ -49,15 +49,15 @@ The following hyperparameters were used during training:
|
|
49 |
|
50 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
51 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
52 |
-
| 0.
|
53 |
-
| 0.
|
54 |
-
| 0.
|
55 |
-
| 0.
|
56 |
|
57 |
|
58 |
### Framework versions
|
59 |
|
60 |
- Transformers 4.37.1
|
61 |
-
- Pytorch 2.1.
|
62 |
- Datasets 2.16.1
|
63 |
- Tokenizers 0.15.1
|
|
|
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 an unknown dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 0.0137
|
21 |
- Accuracy: 1.0
|
22 |
|
23 |
## Model description
|
|
|
49 |
|
50 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
51 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
52 |
+
| 0.4194 | 1.0 | 13 | 0.0737 | 1.0 |
|
53 |
+
| 0.063 | 2.0 | 26 | 0.0266 | 1.0 |
|
54 |
+
| 0.051 | 3.0 | 39 | 0.0151 | 1.0 |
|
55 |
+
| 0.014 | 4.0 | 52 | 0.0137 | 1.0 |
|
56 |
|
57 |
|
58 |
### Framework versions
|
59 |
|
60 |
- Transformers 4.37.1
|
61 |
+
- Pytorch 2.1.0.post103
|
62 |
- Datasets 2.16.1
|
63 |
- Tokenizers 0.15.1
|
all_results.json
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
{
|
2 |
"epoch": 4.0,
|
3 |
-
"train_loss": 0.
|
4 |
-
"train_runtime":
|
5 |
-
"train_samples_per_second":
|
6 |
-
"train_steps_per_second": 0.
|
7 |
}
|
|
|
1 |
{
|
2 |
"epoch": 4.0,
|
3 |
+
"train_loss": 0.11166981543199374,
|
4 |
+
"train_runtime": 294.988,
|
5 |
+
"train_samples_per_second": 2.807,
|
6 |
+
"train_steps_per_second": 0.176
|
7 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 343223968
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7c63a80ea478d192a254c72090c1d314539ba1b3014c9a59bbe9e97f3e5c924d
|
3 |
size 343223968
|
train_results.json
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
{
|
2 |
"epoch": 4.0,
|
3 |
-
"train_loss": 0.
|
4 |
-
"train_runtime":
|
5 |
-
"train_samples_per_second":
|
6 |
-
"train_steps_per_second": 0.
|
7 |
}
|
|
|
1 |
{
|
2 |
"epoch": 4.0,
|
3 |
+
"train_loss": 0.11166981543199374,
|
4 |
+
"train_runtime": 294.988,
|
5 |
+
"train_samples_per_second": 2.807,
|
6 |
+
"train_steps_per_second": 0.176
|
7 |
}
|
trainer_state.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"best_metric": 0.
|
3 |
"best_model_checkpoint": "cat_ds/checkpoint-52",
|
4 |
"epoch": 4.0,
|
5 |
"eval_steps": 100,
|
@@ -11,77 +11,77 @@
|
|
11 |
{
|
12 |
"epoch": 0.77,
|
13 |
"learning_rate": 0.00016153846153846155,
|
14 |
-
"loss": 0.
|
15 |
"step": 10
|
16 |
},
|
17 |
{
|
18 |
"epoch": 1.0,
|
19 |
"eval_accuracy": 1.0,
|
20 |
-
"eval_loss": 0.
|
21 |
-
"eval_runtime":
|
22 |
-
"eval_samples_per_second":
|
23 |
-
"eval_steps_per_second": 0.
|
24 |
"step": 13
|
25 |
},
|
26 |
{
|
27 |
"epoch": 1.54,
|
28 |
"learning_rate": 0.0001230769230769231,
|
29 |
-
"loss": 0.
|
30 |
"step": 20
|
31 |
},
|
32 |
{
|
33 |
"epoch": 2.0,
|
34 |
"eval_accuracy": 1.0,
|
35 |
-
"eval_loss": 0.
|
36 |
-
"eval_runtime":
|
37 |
-
"eval_samples_per_second":
|
38 |
-
"eval_steps_per_second": 0.
|
39 |
"step": 26
|
40 |
},
|
41 |
{
|
42 |
"epoch": 2.31,
|
43 |
"learning_rate": 8.461538461538461e-05,
|
44 |
-
"loss": 0.
|
45 |
"step": 30
|
46 |
},
|
47 |
{
|
48 |
"epoch": 3.0,
|
49 |
"eval_accuracy": 1.0,
|
50 |
-
"eval_loss": 0.
|
51 |
-
"eval_runtime":
|
52 |
-
"eval_samples_per_second":
|
53 |
-
"eval_steps_per_second": 0.
|
54 |
"step": 39
|
55 |
},
|
56 |
{
|
57 |
"epoch": 3.08,
|
58 |
"learning_rate": 4.615384615384616e-05,
|
59 |
-
"loss": 0.
|
60 |
"step": 40
|
61 |
},
|
62 |
{
|
63 |
"epoch": 3.85,
|
64 |
"learning_rate": 7.692307692307694e-06,
|
65 |
-
"loss": 0.
|
66 |
"step": 50
|
67 |
},
|
68 |
{
|
69 |
"epoch": 4.0,
|
70 |
"eval_accuracy": 1.0,
|
71 |
-
"eval_loss": 0.
|
72 |
-
"eval_runtime":
|
73 |
-
"eval_samples_per_second":
|
74 |
-
"eval_steps_per_second": 0.
|
75 |
"step": 52
|
76 |
},
|
77 |
{
|
78 |
"epoch": 4.0,
|
79 |
"step": 52,
|
80 |
"total_flos": 6.416336740088218e+16,
|
81 |
-
"train_loss": 0.
|
82 |
-
"train_runtime":
|
83 |
-
"train_samples_per_second":
|
84 |
-
"train_steps_per_second": 0.
|
85 |
}
|
86 |
],
|
87 |
"logging_steps": 10,
|
|
|
1 |
{
|
2 |
+
"best_metric": 0.013692905195057392,
|
3 |
"best_model_checkpoint": "cat_ds/checkpoint-52",
|
4 |
"epoch": 4.0,
|
5 |
"eval_steps": 100,
|
|
|
11 |
{
|
12 |
"epoch": 0.77,
|
13 |
"learning_rate": 0.00016153846153846155,
|
14 |
+
"loss": 0.4194,
|
15 |
"step": 10
|
16 |
},
|
17 |
{
|
18 |
"epoch": 1.0,
|
19 |
"eval_accuracy": 1.0,
|
20 |
+
"eval_loss": 0.07365512102842331,
|
21 |
+
"eval_runtime": 5.5846,
|
22 |
+
"eval_samples_per_second": 4.298,
|
23 |
+
"eval_steps_per_second": 0.537,
|
24 |
"step": 13
|
25 |
},
|
26 |
{
|
27 |
"epoch": 1.54,
|
28 |
"learning_rate": 0.0001230769230769231,
|
29 |
+
"loss": 0.063,
|
30 |
"step": 20
|
31 |
},
|
32 |
{
|
33 |
"epoch": 2.0,
|
34 |
"eval_accuracy": 1.0,
|
35 |
+
"eval_loss": 0.026600712910294533,
|
36 |
+
"eval_runtime": 5.1625,
|
37 |
+
"eval_samples_per_second": 4.649,
|
38 |
+
"eval_steps_per_second": 0.581,
|
39 |
"step": 26
|
40 |
},
|
41 |
{
|
42 |
"epoch": 2.31,
|
43 |
"learning_rate": 8.461538461538461e-05,
|
44 |
+
"loss": 0.051,
|
45 |
"step": 30
|
46 |
},
|
47 |
{
|
48 |
"epoch": 3.0,
|
49 |
"eval_accuracy": 1.0,
|
50 |
+
"eval_loss": 0.015066702850162983,
|
51 |
+
"eval_runtime": 4.9529,
|
52 |
+
"eval_samples_per_second": 4.846,
|
53 |
+
"eval_steps_per_second": 0.606,
|
54 |
"step": 39
|
55 |
},
|
56 |
{
|
57 |
"epoch": 3.08,
|
58 |
"learning_rate": 4.615384615384616e-05,
|
59 |
+
"loss": 0.0308,
|
60 |
"step": 40
|
61 |
},
|
62 |
{
|
63 |
"epoch": 3.85,
|
64 |
"learning_rate": 7.692307692307694e-06,
|
65 |
+
"loss": 0.014,
|
66 |
"step": 50
|
67 |
},
|
68 |
{
|
69 |
"epoch": 4.0,
|
70 |
"eval_accuracy": 1.0,
|
71 |
+
"eval_loss": 0.013692905195057392,
|
72 |
+
"eval_runtime": 4.8268,
|
73 |
+
"eval_samples_per_second": 4.972,
|
74 |
+
"eval_steps_per_second": 0.622,
|
75 |
"step": 52
|
76 |
},
|
77 |
{
|
78 |
"epoch": 4.0,
|
79 |
"step": 52,
|
80 |
"total_flos": 6.416336740088218e+16,
|
81 |
+
"train_loss": 0.11166981543199374,
|
82 |
+
"train_runtime": 294.988,
|
83 |
+
"train_samples_per_second": 2.807,
|
84 |
+
"train_steps_per_second": 0.176
|
85 |
}
|
86 |
],
|
87 |
"logging_steps": 10,
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 4664
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:524ce2ef9ca1e27cc37ea90486b3f0499f280abaa57ede2315ab5c908badf848
|
3 |
size 4664
|