nutrition-extractor / config.json
raphael0202's picture
Upload LayoutLMv3ForTokenClassification
552d4df verified
raw
history blame contribute delete
No virus
5.24 kB
{
"_name_or_path": "ds-v5-large",
"architectures": [
"LayoutLMv3ForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"classifier_dropout": null,
"coordinate_size": 171,
"eos_token_id": 2,
"finetuning_task": "ner",
"has_relative_attention_bias": true,
"has_spatial_attention_bias": true,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 1024,
"id2label": {
"0": "O",
"1": "B-ENERGY_KJ_SERVING",
"2": "I-ENERGY_KJ_SERVING",
"3": "B-CARBOHYDRATES_100G",
"4": "I-CARBOHYDRATES_100G",
"5": "B-CHOLESTEROL_SERVING",
"6": "I-CHOLESTEROL_SERVING",
"7": "B-ENERGY_KCAL_100G",
"8": "I-ENERGY_KCAL_100G",
"9": "B-SALT_SERVING",
"10": "I-SALT_SERVING",
"11": "B-SALT_100G",
"12": "I-SALT_100G",
"13": "B-SERVING_SIZE",
"14": "I-SERVING_SIZE",
"15": "B-CALCIUM_100G",
"16": "I-CALCIUM_100G",
"17": "B-SODIUM_SERVING",
"18": "I-SODIUM_SERVING",
"19": "B-FIBER_100G",
"20": "I-FIBER_100G",
"21": "B-IRON_SERVING",
"22": "I-IRON_SERVING",
"23": "B-IRON_100G",
"24": "I-IRON_100G",
"25": "B-POTASSIUM_100G",
"26": "I-POTASSIUM_100G",
"27": "B-CALCIUM_SERVING",
"28": "I-CALCIUM_SERVING",
"29": "B-TRANS_FAT_100G",
"30": "I-TRANS_FAT_100G",
"31": "B-SATURATED_FAT_100G",
"32": "I-SATURATED_FAT_100G",
"33": "B-PROTEINS_SERVING",
"34": "I-PROTEINS_SERVING",
"35": "B-SATURATED_FAT_SERVING",
"36": "I-SATURATED_FAT_SERVING",
"37": "B-VITAMIN_D_100G",
"38": "I-VITAMIN_D_100G",
"39": "B-ENERGY_KJ_100G",
"40": "I-ENERGY_KJ_100G",
"41": "B-FAT_100G",
"42": "I-FAT_100G",
"43": "B-PROTEINS_100G",
"44": "I-PROTEINS_100G",
"45": "B-VITAMIN_D_SERVING",
"46": "I-VITAMIN_D_SERVING",
"47": "B-ADDED_SUGARS_SERVING",
"48": "I-ADDED_SUGARS_SERVING",
"49": "B-CHOLESTEROL_100G",
"50": "I-CHOLESTEROL_100G",
"51": "B-SUGARS_100G",
"52": "I-SUGARS_100G",
"53": "B-CARBOHYDRATES_SERVING",
"54": "I-CARBOHYDRATES_SERVING",
"55": "B-ADDED_SUGARS_100G",
"56": "I-ADDED_SUGARS_100G",
"57": "B-SODIUM_100G",
"58": "I-SODIUM_100G",
"59": "B-FIBER_SERVING",
"60": "I-FIBER_SERVING",
"61": "B-SUGARS_SERVING",
"62": "I-SUGARS_SERVING",
"63": "B-ENERGY_KCAL_SERVING",
"64": "I-ENERGY_KCAL_SERVING",
"65": "B-FAT_SERVING",
"66": "I-FAT_SERVING",
"67": "B-TRANS_FAT_SERVING",
"68": "I-TRANS_FAT_SERVING",
"69": "B-POTASSIUM_SERVING",
"70": "I-POTASSIUM_SERVING"
},
"initializer_range": 0.02,
"input_size": 224,
"intermediate_size": 4096,
"label2id": {
"B-ADDED_SUGARS_100G": 55,
"B-ADDED_SUGARS_SERVING": 47,
"B-CALCIUM_100G": 15,
"B-CALCIUM_SERVING": 27,
"B-CARBOHYDRATES_100G": 3,
"B-CARBOHYDRATES_SERVING": 53,
"B-CHOLESTEROL_100G": 49,
"B-CHOLESTEROL_SERVING": 5,
"B-ENERGY_KCAL_100G": 7,
"B-ENERGY_KCAL_SERVING": 63,
"B-ENERGY_KJ_100G": 39,
"B-ENERGY_KJ_SERVING": 1,
"B-FAT_100G": 41,
"B-FAT_SERVING": 65,
"B-FIBER_100G": 19,
"B-FIBER_SERVING": 59,
"B-IRON_100G": 23,
"B-IRON_SERVING": 21,
"B-POTASSIUM_100G": 25,
"B-POTASSIUM_SERVING": 69,
"B-PROTEINS_100G": 43,
"B-PROTEINS_SERVING": 33,
"B-SALT_100G": 11,
"B-SALT_SERVING": 9,
"B-SATURATED_FAT_100G": 31,
"B-SATURATED_FAT_SERVING": 35,
"B-SERVING_SIZE": 13,
"B-SODIUM_100G": 57,
"B-SODIUM_SERVING": 17,
"B-SUGARS_100G": 51,
"B-SUGARS_SERVING": 61,
"B-TRANS_FAT_100G": 29,
"B-TRANS_FAT_SERVING": 67,
"B-VITAMIN_D_100G": 37,
"B-VITAMIN_D_SERVING": 45,
"I-ADDED_SUGARS_100G": 56,
"I-ADDED_SUGARS_SERVING": 48,
"I-CALCIUM_100G": 16,
"I-CALCIUM_SERVING": 28,
"I-CARBOHYDRATES_100G": 4,
"I-CARBOHYDRATES_SERVING": 54,
"I-CHOLESTEROL_100G": 50,
"I-CHOLESTEROL_SERVING": 6,
"I-ENERGY_KCAL_100G": 8,
"I-ENERGY_KCAL_SERVING": 64,
"I-ENERGY_KJ_100G": 40,
"I-ENERGY_KJ_SERVING": 2,
"I-FAT_100G": 42,
"I-FAT_SERVING": 66,
"I-FIBER_100G": 20,
"I-FIBER_SERVING": 60,
"I-IRON_100G": 24,
"I-IRON_SERVING": 22,
"I-POTASSIUM_100G": 26,
"I-POTASSIUM_SERVING": 70,
"I-PROTEINS_100G": 44,
"I-PROTEINS_SERVING": 34,
"I-SALT_100G": 12,
"I-SALT_SERVING": 10,
"I-SATURATED_FAT_100G": 32,
"I-SATURATED_FAT_SERVING": 36,
"I-SERVING_SIZE": 14,
"I-SODIUM_100G": 58,
"I-SODIUM_SERVING": 18,
"I-SUGARS_100G": 52,
"I-SUGARS_SERVING": 62,
"I-TRANS_FAT_100G": 30,
"I-TRANS_FAT_SERVING": 68,
"I-VITAMIN_D_100G": 38,
"I-VITAMIN_D_SERVING": 46,
"O": 0
},
"layer_norm_eps": 1e-05,
"max_2d_position_embeddings": 1024,
"max_position_embeddings": 514,
"max_rel_2d_pos": 256,
"max_rel_pos": 128,
"model_type": "layoutlmv3",
"num_attention_heads": 16,
"num_channels": 3,
"num_hidden_layers": 24,
"pad_token_id": 1,
"patch_size": 16,
"rel_2d_pos_bins": 64,
"rel_pos_bins": 32,
"second_input_size": 112,
"shape_size": 170,
"text_embed": true,
"torch_dtype": "float32",
"transformers_version": "4.40.2",
"type_vocab_size": 1,
"visual_embed": true,
"vocab_size": 50265
}