Commit
·
f37c32b
1
Parent(s):
ee22bae
Update tiny models for ConvNextV2ForImageClassification
Browse files- config.json +7 -1
- model.safetensors +3 -0
- tf_model.h5 +3 -0
config.json
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
{
|
|
|
2 |
"architectures": [
|
3 |
"ConvNextV2ForImageClassification"
|
4 |
],
|
@@ -51,6 +52,11 @@
|
|
51 |
"stage3",
|
52 |
"stage4"
|
53 |
],
|
|
|
|
|
|
|
|
|
|
|
54 |
"patch_size": 4,
|
55 |
"stage_names": [
|
56 |
"stem",
|
@@ -60,5 +66,5 @@
|
|
60 |
"stage4"
|
61 |
],
|
62 |
"torch_dtype": "float32",
|
63 |
-
"transformers_version": "4.
|
64 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "tiny_models/convnextv2/ConvNextV2ForImageClassification",
|
3 |
"architectures": [
|
4 |
"ConvNextV2ForImageClassification"
|
5 |
],
|
|
|
52 |
"stage3",
|
53 |
"stage4"
|
54 |
],
|
55 |
+
"out_indices": [
|
56 |
+
2,
|
57 |
+
3,
|
58 |
+
4
|
59 |
+
],
|
60 |
"patch_size": 4,
|
61 |
"stage_names": [
|
62 |
"stem",
|
|
|
66 |
"stage4"
|
67 |
],
|
68 |
"torch_dtype": "float32",
|
69 |
+
"transformers_version": "4.36.0.dev0"
|
70 |
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4d0b7525a179a541c9795dd24e0d21161b0c55ac901c6c9a599d84ea4000bc94
|
3 |
+
size 329856
|
tf_model.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:176c6140f71367343be19f9f134beb8e5a6a18699f4c998bc7f75f6a4345a5de
|
3 |
+
size 442368
|