carlosleao commited on
Commit
a710584
1 Parent(s): 5f2e7e6

Training in progress, step 500

Browse files
Files changed (3) hide show
  1. config.json +16 -16
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "carlosleao/vit-Facial-Expression-Recognition",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -9,25 +9,25 @@
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
  },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
- "anger": 4,
25
- "disgust": 5,
26
- "fear": 6,
27
- "happiness": 1,
28
- "neutral": 0,
29
- "sadness": 3,
30
- "surprise": 2
31
  },
32
  "layer_norm_eps": 1e-12,
33
  "model_type": "vit",
@@ -38,5 +38,5 @@
38
  "problem_type": "single_label_classification",
39
  "qkv_bias": true,
40
  "torch_dtype": "float32",
41
- "transformers_version": "4.45.1"
42
  }
 
1
  {
2
+ "_name_or_path": "motheecreator/vit-Facial-Expression-Recognition",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "Surprise",
13
+ "1": "Fear",
14
+ "2": "Disgust",
15
+ "3": "Happiness",
16
+ "4": "Sadness",
17
+ "5": "Anger",
18
+ "6": "Neutral"
19
  },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
+ "Anger": 5,
25
+ "Disgust": 2,
26
+ "Fear": 1,
27
+ "Happiness": 3,
28
+ "Neutral": 6,
29
+ "Sadness": 4,
30
+ "Surprise": 0
31
  },
32
  "layer_norm_eps": 1e-12,
33
  "model_type": "vit",
 
38
  "problem_type": "single_label_classification",
39
  "qkv_bias": true,
40
  "torch_dtype": "float32",
41
+ "transformers_version": "4.45.2"
42
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f3d04aac547dd1d00b80e96b938740837ae18c48abffefecb2fc3a62d4b86779
3
  size 343239356
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3e7c08a4ea8b49211ea9e8228ee443f9626ced7313143b550a390bce37a8573
3
  size 343239356
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c9256a3ca99a9d02f853db70846c10bdaa72c82c2ba9057d109cabc31b536696
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b99c9e93c970b38db62b2c77d500c22a77c99a6701c3c7ef93e927a50a91d2bc
3
  size 5304