Update tiny models for ConvNextV2ForImageClassification

#17
Files changed (3) hide show
  1. config.json +7 -1
  2. model.safetensors +3 -0
  3. 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.28.0.dev0"
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:013f840d225bf4fcba7697761d132ba4843b69848ea162da7ecb0df6406dd27b
3
+ size 329856
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab70ceb470926d46f928eba05a7052642607be26a140fd40f2bb39dd52d94b5f
3
+ size 442368