fgrezes's picture
fixed sized mismatch
62881d6 verified
{
"_name_or_path": "../../../astroBERT-Tasks/Finetuning_1_NER/trained-models/INDUS_003/checkpoint-3500/",
"architectures": [
"RobertaForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"classifier_dropout": null,
"eos_token_id": 2,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "B-Archive",
"1": "B-CelestialObject",
"2": "B-CelestialObjectRegion",
"3": "B-CelestialRegion",
"4": "B-Citation",
"5": "B-Collaboration",
"6": "B-ComputingFacility",
"7": "B-Database",
"8": "B-Dataset",
"9": "B-EntityOfFutureInterest",
"10": "B-Event",
"11": "B-Fellowship",
"12": "B-Formula",
"13": "B-Grant",
"14": "B-Identifier",
"15": "B-Instrument",
"16": "B-Location",
"17": "B-Mission",
"18": "B-Model",
"19": "B-ObservationalTechniques",
"20": "B-Observatory",
"21": "B-Organization",
"22": "B-Person",
"23": "B-Proposal",
"24": "B-Software",
"25": "B-Survey",
"26": "B-Tag",
"27": "B-Telescope",
"28": "B-TextGarbage",
"29": "B-URL",
"30": "B-Wavelength",
"31": "I-Archive",
"32": "I-CelestialObject",
"33": "I-CelestialObjectRegion",
"34": "I-CelestialRegion",
"35": "I-Citation",
"36": "I-Collaboration",
"37": "I-ComputingFacility",
"38": "I-Database",
"39": "I-Dataset",
"40": "I-EntityOfFutureInterest",
"41": "I-Event",
"42": "I-Fellowship",
"43": "I-Formula",
"44": "I-Grant",
"45": "I-Identifier",
"46": "I-Instrument",
"47": "I-Location",
"48": "I-Mission",
"49": "I-Model",
"50": "I-ObservationalTechniques",
"51": "I-Observatory",
"52": "I-Organization",
"53": "I-Person",
"54": "I-Proposal",
"55": "I-Software",
"56": "I-Survey",
"57": "I-Tag",
"58": "I-Telescope",
"59": "I-TextGarbage",
"60": "I-URL",
"61": "I-Wavelength",
"62": "O",
"63": "Special_Token"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"B-Archive": 0,
"B-CelestialObject": 1,
"B-CelestialObjectRegion": 2,
"B-CelestialRegion": 3,
"B-Citation": 4,
"B-Collaboration": 5,
"B-ComputingFacility": 6,
"B-Database": 7,
"B-Dataset": 8,
"B-EntityOfFutureInterest": 9,
"B-Event": 10,
"B-Fellowship": 11,
"B-Formula": 12,
"B-Grant": 13,
"B-Identifier": 14,
"B-Instrument": 15,
"B-Location": 16,
"B-Mission": 17,
"B-Model": 18,
"B-ObservationalTechniques": 19,
"B-Observatory": 20,
"B-Organization": 21,
"B-Person": 22,
"B-Proposal": 23,
"B-Software": 24,
"B-Survey": 25,
"B-Tag": 26,
"B-Telescope": 27,
"B-TextGarbage": 28,
"B-URL": 29,
"B-Wavelength": 30,
"I-Archive": 31,
"I-CelestialObject": 32,
"I-CelestialObjectRegion": 33,
"I-CelestialRegion": 34,
"I-Citation": 35,
"I-Collaboration": 36,
"I-ComputingFacility": 37,
"I-Database": 38,
"I-Dataset": 39,
"I-EntityOfFutureInterest": 40,
"I-Event": 41,
"I-Fellowship": 42,
"I-Formula": 43,
"I-Grant": 44,
"I-Identifier": 45,
"I-Instrument": 46,
"I-Location": 47,
"I-Mission": 48,
"I-Model": 49,
"I-ObservationalTechniques": 50,
"I-Observatory": 51,
"I-Organization": 52,
"I-Person": 53,
"I-Proposal": 54,
"I-Software": 55,
"I-Survey": 56,
"I-Tag": 57,
"I-Telescope": 58,
"I-TextGarbage": 59,
"I-URL": 60,
"I-Wavelength": 61,
"O": 62,
"Special_Token": 63
},
"layer_norm_eps": 1e-05,
"max_position_embeddings": 514,
"model_type": "roberta",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 1,
"position_embedding_type": "absolute",
"torch_dtype": "float32",
"transformers_version": "4.46.2",
"type_vocab_size": 1,
"use_cache": true,
"vocab_size": 50265
}