carlosleao commited on
Commit
3746f32
1 Parent(s): fd1bb01

Training in progress, step 500

Browse files
Files changed (3) hide show
  1. config.json +17 -15
  2. model.safetensors +2 -2
  3. training_args.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "motheecreator/vit-Facial-Expression-Recognition",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -9,25 +9,27 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "angry",
13
- "1": "disgust",
14
- "2": "fear",
15
- "3": "happy",
16
- "4": "neutral",
17
- "5": "sad",
18
- "6": "surprise"
 
19
  },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
- "angry": 0,
25
- "disgust": 1,
26
- "fear": 2,
27
- "happy": 3,
28
- "neutral": 4,
29
- "sad": 5,
30
- "surprise": 6
 
31
  },
32
  "layer_norm_eps": 1e-12,
33
  "model_type": "vit",
 
1
  {
2
+ "_name_or_path": "carlosleao/vit-Facial-Expression-Recognition",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "neutral",
13
+ "1": "happiness",
14
+ "2": "surprise",
15
+ "3": "sadness",
16
+ "4": "anger",
17
+ "5": "disgust",
18
+ "6": "fear",
19
+ "7": "contempt"
20
  },
21
  "image_size": 224,
22
  "initializer_range": 0.02,
23
  "intermediate_size": 3072,
24
  "label2id": {
25
+ "anger": 4,
26
+ "contempt": 7,
27
+ "disgust": 5,
28
+ "fear": 6,
29
+ "happiness": 1,
30
+ "neutral": 0,
31
+ "sadness": 3,
32
+ "surprise": 2
33
  },
34
  "layer_norm_eps": 1e-12,
35
  "model_type": "vit",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7efd64c38d3d9748f324ce7bea50a10a694d7bacd2a913b0ee4c845194fa2abd
3
- size 343239356
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:480e46cd317a849f3e54cf0bd2f6f140e77ea65bfd47f996e4df5b2001ea5113
3
+ size 343242432
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:96f04a8da32ec847a1c357f108f60bcb5d0f6ffe0ae01fe9298683158d26c94f
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bedb9432edb0bf54907d5430c04fdf36db1baa32be2e6414a587af29a0eb0256
3
  size 5240