yehiawp4 commited on
Commit
b707b0f
1 Parent(s): 3df9603

Training in progress, epoch 0

Browse files
config.json CHANGED
@@ -13,20 +13,24 @@
13
  "hidden_size": 768,
14
  "id2label": {
15
  "0": "Anger",
16
- "1": "Happy",
17
- "2": "Neutral",
18
- "3": "Sad",
19
- "4": "Surprise"
 
 
20
  },
21
  "image_size": 224,
22
  "initializer_range": 0.02,
23
  "intermediate_size": 3072,
24
  "label2id": {
25
  "Anger": 0,
26
- "Happy": 1,
27
- "Neutral": 2,
28
- "Sad": 3,
29
- "Surprise": 4
 
 
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "model_type": "videomae",
@@ -39,7 +43,7 @@
39
  "problem_type": "single_label_classification",
40
  "qkv_bias": true,
41
  "torch_dtype": "float32",
42
- "transformers_version": "4.38.2",
43
  "tubelet_size": 2,
44
  "use_mean_pooling": true
45
  }
 
13
  "hidden_size": 768,
14
  "id2label": {
15
  "0": "Anger",
16
+ "1": "Disgust",
17
+ "2": "Fear",
18
+ "3": "Happy",
19
+ "4": "Neutral",
20
+ "5": "Sad",
21
+ "6": "Surprise"
22
  },
23
  "image_size": 224,
24
  "initializer_range": 0.02,
25
  "intermediate_size": 3072,
26
  "label2id": {
27
  "Anger": 0,
28
+ "Disgust": 1,
29
+ "Fear": 2,
30
+ "Happy": 3,
31
+ "Neutral": 4,
32
+ "Sad": 5,
33
+ "Surprise": 6
34
  },
35
  "layer_norm_eps": 1e-12,
36
  "model_type": "videomae",
 
43
  "problem_type": "single_label_classification",
44
  "qkv_bias": true,
45
  "torch_dtype": "float32",
46
+ "transformers_version": "4.39.0",
47
  "tubelet_size": 2,
48
  "use_mean_pooling": true
49
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:723a6f38ed09850aeec103381e1b0842d774cbaf739093a95fd9c130d10824bc
3
- size 344946564
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cfc2018d5a874ef16c1ca26d894a1f842ef30e987f33c2d20ca35a10ed5bce2
3
+ size 344952716
preprocessor_config.json CHANGED
@@ -1,4 +1,20 @@
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "crop_size": {
3
  "height": 224,
4
  "width": 224
 
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
+ "do_normalize",
12
+ "image_mean",
13
+ "image_std",
14
+ "return_tensors",
15
+ "data_format",
16
+ "input_data_format"
17
+ ],
18
  "crop_size": {
19
  "height": 224,
20
  "width": 224
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:39abc1cad149b0252ccc0bfbc57825bebc22e5c9ec8cb465cc284234a9cebbaa
3
- size 4984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:202de3e04bb4d7ddf9657960358d9cf11dd0bce6f142c896f871be4f59feebce
3
+ size 5048