Amirhossein75 commited on
Commit
5fe86f6
·
verified ·
1 Parent(s): 93f0f2c

Upload inference_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. inference_config.json +22 -0
inference_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "encoder_name": "openai/clip-vit-base-patch32",
3
+ "backend": "clip",
4
+ "head": "mtl",
5
+ "fusion_dim": 512,
6
+ "threshold": 0.5,
7
+ "thresholds": [
8
+ 0.3,
9
+ 0.7,
10
+ 0.49999999999999994,
11
+ 0.3,
12
+ 0.5499999999999999
13
+ ],
14
+ "class_names": [
15
+ "racist",
16
+ "sexist",
17
+ "homophobe",
18
+ "religion",
19
+ "otherhate"
20
+ ],
21
+ "best_checkpoint_dir": "runs/clip_mtl_mmhshateful\\checkpoint-11236"
22
+ }