bert_model / config.json
DRAGOO's picture
Update config.json
da3443d verified
raw
history blame
2.88 kB
{
"_name_or_path": "distilbert-base-uncased",
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0" : "AIR COMPRESSOR" ,
"1" : "ALLEN KEY" ,
"2" : "ANGLE GAUGE" ,
"3" : "AXE" ,
"4" : "BACK SAW" ,
"5" : "BAND SAW" ,
"6" : "BONE SAW" ,
"7" : "CALIPER" ,
"8" : "CHISEL" ,
"9" : "CIRCULAR SAW" ,
"10" : "CLAMP" ,
"11" : "COPING SAW" ,
"12" : "CORDLESS DRILL" ,
"13" : "CROSSCUT SAW" ,
"14" : "DRILLS" ,
"15" : "FILE" ,
"16" : "FRET SAW" ,
"17" : "GARAGE CABINET" ,
"18" : "GAUGE" ,
"19" : "GRINDER" ,
"20" : "HACK SAW" ,
"21" : "HAMMER" ,
"22" : "HAND SAW" ,
"23" : "HEAT GUN" ,
"24" : "IMPACT DRIVER" ,
"25" : "IMPACT WRENCH" ,
"26" : "INCLINOMETER" ,
"27" : "KEYHOLE SAW" ,
"28" : "LASER LEVEL" ,
"29" : "LATHE" ,
"30" : "LEVEL" ,
"31" : "MALLET" ,
"32" : "MICROMETER" ,
"33" : "MITER SAW" ,
"34" : "NAIL GUN" ,
"35" : "NUT DRIVER" ,
"36" : "PLANE" ,
"37" : "PLIERS" ,
"38" : "PROTRACTOR" ,
"39" : "PRUNING SAW" ,
"40" : "RASP" ,
"41" : "RECIPROCATING SAW" ,
"42" : "ROLLING TOOL STORAGE" ,
"43" : "SANDER" ,
"44" : "SCREW DRIVERS" ,
"45" : "STAPLER" ,
"46" : "TABLE SAW" ,
"47" : "TAPE MEASURE" ,
"48" : "TILE SAW" ,
"49" : "TOOL BAG" ,
"50" : "TOOL BELT" ,
"51" : "TOOL BOX" ,
"52" : "TOOL CABINET" ,
"53" : "TOOL CHEST" ,
"54" : "VENEER SAW" ,
"55" : "WRENCH"
},
"initializer_range": 0.02,
"label2id": {
"AIR COMPRESSOR" : 0 ,
"ALLEN KEY" : 1 ,
"ANGLE GAUGE" : 2 ,
"AXE" : 3 ,
"BACK SAW" : 4 ,
"BAND SAW" : 5 ,
"BONE SAW" : 6 ,
"CALIPER" : 7 ,
"CHISEL" : 8 ,
"CIRCULAR SAW" : 9 ,
"CLAMP" : 10 ,
"COPING SAW" : 11 ,
"CORDLESS DRILL" : 12 ,
"CROSSCUT SAW" : 13 ,
"DRILLS" : 14 ,
"FILE" : 15 ,
"FRET SAW" : 16 ,
"GARAGE CABINET" : 17 ,
"GAUGE" : 18 ,
"GRINDER" : 19 ,
"HACK SAW" : 20 ,
"HAMMER" : 21 ,
"HAND SAW" : 22 ,
"HEAT GUN" : 23 ,
"IMPACT DRIVER" : 24 ,
"IMPACT WRENCH" : 25 ,
"INCLINOMETER" : 26 ,
"KEYHOLE SAW" : 27 ,
"LASER LEVEL" : 28 ,
"LATHE" : 29 ,
"LEVEL" : 30 ,
"MALLET" : 31 ,
"MICROMETER" : 32 ,
"MITER SAW" : 33 ,
"NAIL GUN" : 34 ,
"NUT DRIVER" : 35 ,
"PLANE" : 36 ,
"PLIERS" : 37 ,
"PROTRACTOR" : 38 ,
"PRUNING SAW" : 39 ,
"RASP" : 40 ,
"RECIPROCATING SAW" : 41 ,
"ROLLING TOOL STORAGE" : 42 ,
"SANDER" : 43 ,
"SCREW DRIVERS" : 44 ,
"STAPLER" : 45 ,
"TABLE SAW" : 46 ,
"TAPE MEASURE" : 47 ,
"TILE SAW" : 48 ,
"TOOL BAG" : 49 ,
"TOOL BELT" : 50 ,
"TOOL BOX" : 51 ,
"TOOL CABINET" : 52 ,
"TOOL CHEST" : 53 ,
"VENEER SAW" : 54 ,
"WRENCH" : 55
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.40.0",
"vocab_size": 30522
}