Upload 4 files
Browse filescorrect initial model
- config.json +2 -3
- model.safetensors +1 -1
- training_args.bin +3 -0
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"ViTForImageClassification"
|
5 |
],
|
@@ -65,8 +65,7 @@
|
|
65 |
"num_channels": 3,
|
66 |
"num_hidden_layers": 12,
|
67 |
"patch_size": 16,
|
68 |
-
"problem_type": "single_label_classification",
|
69 |
"qkv_bias": true,
|
70 |
"torch_dtype": "float32",
|
71 |
-
"transformers_version": "4.
|
72 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "google/vit-base-patch16-224-in21k",
|
3 |
"architectures": [
|
4 |
"ViTForImageClassification"
|
5 |
],
|
|
|
65 |
"num_channels": 3,
|
66 |
"num_hidden_layers": 12,
|
67 |
"patch_size": 16,
|
|
|
68 |
"qkv_bias": true,
|
69 |
"torch_dtype": "float32",
|
70 |
+
"transformers_version": "4.46.3"
|
71 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 343285496
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1d2cfc40baa5f67df7536485b19c49a914932668222666dad945621fd07defa0
|
3 |
size 343285496
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3d3456005124c0de74e415ad33d431c52e27406ad11f70704107b3cf495d49cd
|
3 |
+
size 5240
|