sanali209 commited on
Commit
7d15377
·
verified ·
1 Parent(s): 8d6598b

Upload best model after training (val_loss: 0.4664)

Browse files
Files changed (2) hide show
  1. config.json +2 -4
  2. model.safetensors +2 -2
config.json CHANGED
@@ -11,8 +11,7 @@
11
  "id2label": {
12
  "0": "explicit",
13
  "1": "general",
14
- "2": "questionable",
15
- "3": "sensitive"
16
  },
17
  "image_size": 224,
18
  "initializer_range": 0.02,
@@ -20,8 +19,7 @@
20
  "label2id": {
21
  "explicit": 0,
22
  "general": 1,
23
- "questionable": 2,
24
- "sensitive": 3
25
  },
26
  "layer_norm_eps": 1e-12,
27
  "model_type": "vit",
 
11
  "id2label": {
12
  "0": "explicit",
13
  "1": "general",
14
+ "2": "sensitive"
 
15
  },
16
  "image_size": 224,
17
  "initializer_range": 0.02,
 
19
  "label2id": {
20
  "explicit": 0,
21
  "general": 1,
22
+ "sensitive": 2
 
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "model_type": "vit",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:255c7b851f4a4151c49e79fdf424785d65348d31b1c587e84d8e94005a277f65
3
- size 343230128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ab7e9771e7b65ad3ed5c2c8d128b3f6e21bc87a0e4ab0f970e69ef97e0e16e2
3
+ size 343227052