Upload 8 files
Browse files- config.json +51 -0
- model.safetensors +3 -0
- optimizer.pt +3 -0
- preprocessor_config.json +45 -0
- rng_state.pth +3 -0
- scheduler.pt +3 -0
- trainer_state.json +0 -0
- training_args.bin +3 -0
config.json
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "google/vivit-b-16x2",
|
3 |
+
"architectures": [
|
4 |
+
"VivitForVideoClassification"
|
5 |
+
],
|
6 |
+
"attention_probs_dropout_prob": 0.0,
|
7 |
+
"hidden_act": "gelu_fast",
|
8 |
+
"hidden_dropout_prob": 0.0,
|
9 |
+
"hidden_size": 768,
|
10 |
+
"id2label": {
|
11 |
+
"0": "finish",
|
12 |
+
"1": "fish",
|
13 |
+
"2": "forget",
|
14 |
+
"3": "france",
|
15 |
+
"4": "friend",
|
16 |
+
"5": "girl",
|
17 |
+
"6": "good",
|
18 |
+
"7": "grandfather",
|
19 |
+
"8": "grandmother",
|
20 |
+
"9": "green"
|
21 |
+
},
|
22 |
+
"image_size": 224,
|
23 |
+
"initializer_range": 0.02,
|
24 |
+
"intermediate_size": 3072,
|
25 |
+
"label2id": {
|
26 |
+
"finish": 0,
|
27 |
+
"fish": 1,
|
28 |
+
"forget": 2,
|
29 |
+
"france": 3,
|
30 |
+
"friend": 4,
|
31 |
+
"girl": 5,
|
32 |
+
"good": 6,
|
33 |
+
"grandfather": 7,
|
34 |
+
"grandmother": 8,
|
35 |
+
"green": 9
|
36 |
+
},
|
37 |
+
"layer_norm_eps": 1e-06,
|
38 |
+
"model_type": "vivit",
|
39 |
+
"num_attention_heads": 12,
|
40 |
+
"num_channels": 3,
|
41 |
+
"num_frames": 32,
|
42 |
+
"num_hidden_layers": 12,
|
43 |
+
"qkv_bias": true,
|
44 |
+
"torch_dtype": "float32",
|
45 |
+
"transformers_version": "4.40.0",
|
46 |
+
"tubelet_size": [
|
47 |
+
2,
|
48 |
+
16,
|
49 |
+
16
|
50 |
+
]
|
51 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:03583bbf3d6fb19303ee2b5d6b9cadac324b28f8f1f107a08b61f088d7208abe
|
3 |
+
size 354639992
|
optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aa84351a6df7d663664115fcdc8f93d6e32f4392a7c66e85da6a6d2332527a57
|
3 |
+
size 709399994
|
preprocessor_config.json
ADDED
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_valid_processor_keys": [
|
3 |
+
"videos",
|
4 |
+
"do_resize",
|
5 |
+
"size",
|
6 |
+
"resample",
|
7 |
+
"do_center_crop",
|
8 |
+
"crop_size",
|
9 |
+
"do_rescale",
|
10 |
+
"rescale_factor",
|
11 |
+
"offset",
|
12 |
+
"do_normalize",
|
13 |
+
"image_mean",
|
14 |
+
"image_std",
|
15 |
+
"return_tensors",
|
16 |
+
"data_format",
|
17 |
+
"input_data_format"
|
18 |
+
],
|
19 |
+
"crop_size": {
|
20 |
+
"height": 224,
|
21 |
+
"width": 224
|
22 |
+
},
|
23 |
+
"do_center_crop": true,
|
24 |
+
"do_normalize": false,
|
25 |
+
"do_rescale": true,
|
26 |
+
"do_resize": true,
|
27 |
+
"do_zero_centering": true,
|
28 |
+
"image_mean": [
|
29 |
+
0.5,
|
30 |
+
0.5,
|
31 |
+
0.5
|
32 |
+
],
|
33 |
+
"image_processor_type": "VivitImageProcessor",
|
34 |
+
"image_std": [
|
35 |
+
0.5,
|
36 |
+
0.5,
|
37 |
+
0.5
|
38 |
+
],
|
39 |
+
"offset": true,
|
40 |
+
"resample": 2,
|
41 |
+
"rescale_factor": 0.00784313725490196,
|
42 |
+
"size": {
|
43 |
+
"shortest_edge": 256
|
44 |
+
}
|
45 |
+
}
|
rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e6b70d5754170e7f498ad12616a63f7a333f7839adc7faa68372059d06354f81
|
3 |
+
size 14244
|
scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:83246533b7d2076cc802e0402ff73ac1f8cff7ed1ded7f015c6d134e43c19415
|
3 |
+
size 1064
|
trainer_state.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6d7583198c57f43cb0b7d69cc71e8c1c4eac9335f73d802013c8148fe01d8fdf
|
3 |
+
size 5048
|