jegormeister commited on
Commit
e3bcd5e
1 Parent(s): ef101d1

Add ViViT model weights

Browse files
Files changed (3) hide show
  1. config.json +29 -0
  2. preprocessor_config.json +27 -0
  3. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViViTForVideoClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "hidden_act": "gelu_fast",
7
+ "hidden_dropout_prob": 0.0,
8
+ "hidden_size": 768,
9
+ "initializer_range": 0.02,
10
+ "intermediate_size": 3072,
11
+ "layer_norm_eps": 1e-06,
12
+ "model_type": "vivit",
13
+ "num_attention_heads": 12,
14
+ "num_channels": 3,
15
+ "num_hidden_layers": 12,
16
+ "qkv_bias": true,
17
+ "torch_dtype": "float32",
18
+ "transformers_version": "4.25.0.dev0",
19
+ "tubelet_size": [
20
+ 2,
21
+ 16,
22
+ 16
23
+ ],
24
+ "video_size": [
25
+ 32,
26
+ 224,
27
+ 224
28
+ ]
29
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_normalize": true,
8
+ "do_rescale": true,
9
+ "do_resize": true,
10
+ "image_mean": [
11
+ 0.5,
12
+ 0.5,
13
+ 0.5
14
+ ],
15
+ "image_processor_type": "ViViTImageProcessor",
16
+ "image_std": [
17
+ 0.5,
18
+ 0.5,
19
+ 0.5
20
+ ],
21
+ "resample": 2,
22
+ "rescale_factor": 0.00392156862745098,
23
+ "size": {
24
+ "shortest_edge": 224
25
+ }
26
+ }
27
+
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0501b48c0dcef46c494e78a296a9b85310570cda99b4727d4dc869804078dcd4
3
+ size 354657325