gullalc commited on
Commit
638b681
1 Parent(s): d60ef61

Upload VideoMAEForVideoClassification

Browse files
Files changed (2) hide show
  1. config.json +9 -11
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "models/movienet_scale/MCG-NJU/videomae-base-finetuned-kinetics",
3
  "architectures": [
4
  "VideoMAEForVideoClassification"
5
  ],
@@ -12,21 +12,19 @@
12
  "hidden_dropout_prob": 0.0,
13
  "hidden_size": 768,
14
  "id2label": {
15
- "0": "ECS",
16
- "1": "CS",
17
- "2": "MS",
18
- "3": "FS",
19
- "4": "LS"
20
  },
21
  "image_size": 224,
22
  "initializer_range": 0.02,
23
  "intermediate_size": 3072,
24
  "label2id": {
25
- "CS": 1,
26
- "ECS": 0,
27
- "FS": 3,
28
- "LS": 4,
29
- "MS": 2
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "model_type": "videomae",
 
1
  {
2
+ "_name_or_path": "models/movienet_movement/MCG-NJU/videomae-base-finetuned-kinetics",
3
  "architectures": [
4
  "VideoMAEForVideoClassification"
5
  ],
 
12
  "hidden_dropout_prob": 0.0,
13
  "hidden_size": 768,
14
  "id2label": {
15
+ "0": "Static",
16
+ "1": "Motion",
17
+ "2": "Pull",
18
+ "3": "Push"
 
19
  },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
+ "Motion": 1,
25
+ "Pull": 2,
26
+ "Push": 3,
27
+ "Static": 0
 
28
  },
29
  "layer_norm_eps": 1e-12,
30
  "model_type": "videomae",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cf9191245db54628209e42bbe8e278b6a8898f14be6c545dbde10c193ca6bd84
3
- size 344946564
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d8241b62e0d0f131b104c3235beb4ea28a4f32bd4c1b39d16325e7a9f029ef6
3
+ size 344943488