Upload RelationPT_rob

config.json

@@ -1,98 +1,85 @@
 {
-  "_name_or_path": "./train_pret_bert_cased_3101_05/checkpoint-839",
+  "_name_or_path": "./train_pret_cos_random/checkpoint-599",
   "architectures": [
-    "RelationPT"
+    "RelationPT_rob"
   ],
   "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
   "classifier_dropout": null,
   "dec_hidden_layers": 1,
-  "gradient_checkpointing": false,
+  "eos_token_id": 2,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
-  "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4"
-  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
-  "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4
-  },
-  "layer_norm_eps": 1e-12,
-  "max_position_embeddings": 512,
-  "model_type": "bert",
+  "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 514,
+  "model_type": "roberta",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
-  "pad_token_id": 0,
+  "pad_token_id": 1,
   "position_embedding_type": "absolute",
   "torch_dtype": "float32",
   "transformers_version": "4.24.0",
-  "type_vocab_size": 2,
+  "type_vocab_size": 1,
   "use_cache": true,
   "vocab": {
     "attribute2id": {
       "<PAD>": 0,
-      "altitude": 53,
-      "arg_of_periapsis": 40,
-      "azimuths": 29,
-      "burn_time": 19,
-      "comment": 36,
-      "constraint": 23,
-      "cospar_id": 45,
-      "cospar_launch_number": 51,
-      "depth": 13,
-      "diameter": 8,
-      "discos_id": 42,
-      "dry_mass": 38,
-      "eccentricity": 18,
-      "epoch": 3,
-      "esc_capacity": 4,
-      "failed_launches": 2,
-      "failure": 5,
-      "flight_number": 6,
-      "fragmentation_event_type": 37,
-      "fuel": 27,
-      "fuel_mass": 24,
-      "geo_capacity": 25,
-      "gto_capacity": 21,
-      "height": 1,
-      "height_of_centre": 20,
-      "height_of_pericentre": 33,
-      "inclination": 12,
-      "latitude": 34,
-      "launch_pad": 7,
-      "leo_capacity": 30,
-      "longitude": 41,
-      "mass": 48,
-      "max_cross_section": 14,
-      "max_specific_impulse": 50,
-      "mean_anomaly": 22,
-      "min_cross_section": 32,
-      "number_of_stages": 44,
-      "oxidiser": 17,
-      "oxidiser_mass": 31,
-      "raan": 43,
-      "reentry": 47,
-      "satno": 39,
-      "semimajor_axis": 9,
-      "shape": 52,
-      "solid_propellant": 46,
-      "solid_propellant_mass": 11,
-      "span": 16,
-      "sso_capacity": 26,
-      "successful_launches": 35,
-      "thrust_level": 10,
-      "vimpel_id": 28,
-      "wet_mass": 15,
-      "width": 49
+      "altitude": 4,
+      "arg_of_periapsis": 30,
+      "azimuths": 36,
+      "burn_time": 2,
+      "comment": 18,
+      "constraint": 49,
+      "cospar_id": 52,
+      "cospar_launch_number": 21,
+      "depth": 27,
+      "diameter": 12,
+      "discos_id": 48,
+      "dry_mass": 35,
+      "eccentricity": 41,
+      "epoch": 10,
+      "esc_capacity": 34,
+      "failed_launches": 33,
+      "failure": 43,
+      "flight_number": 15,
+      "fragmentation_event_type": 42,
+      "fuel": 6,
+      "fuel_mass": 26,
+      "geo_capacity": 40,
+      "gto_capacity": 31,
+      "height": 22,
+      "height_of_centre": 24,
+      "height_of_pericentre": 17,
+      "inclination": 50,
+      "latitude": 20,
+      "launch_pad": 5,
+      "leo_capacity": 39,
+      "longitude": 7,
+      "mass": 14,
+      "max_cross_section": 23,
+      "max_specific_impulse": 37,
+      "mean_anomaly": 46,
+      "min_cross_section": 13,
+      "number_of_stages": 1,
+      "oxidiser": 38,
+      "oxidiser_mass": 29,
+      "raan": 3,
+      "reentry": 19,
+      "satno": 51,
+      "semimajor_axis": 16,
+      "shape": 53,
+      "solid_propellant": 47,
+      "solid_propellant_mass": 9,
+      "span": 44,
+      "sso_capacity": 28,
+      "successful_launches": 11,
+      "thrust_level": 45,
+      "vimpel_id": 32,
+      "wet_mass": 8,
+      "width": 25
     },
     "concept2id": {
       "<PAD>": 0,
@@ -91250,59 +91237,59 @@
     },
     "id2attribute": [
       "<PAD>",
-      "height",
-      "failed_launches",
-      "epoch",
-      "esc_capacity",
-      "failure",
-      "flight_number",
+      "number_of_stages",
+      "burn_time",
+      "raan",
+      "altitude",
       "launch_pad",
+      "fuel",
+      "longitude",
+      "wet_mass",
+      "solid_propellant_mass",
+      "epoch",
+      "successful_launches",
       "diameter",
+      "min_cross_section",
+      "mass",
+      "flight_number",
       "semimajor_axis",
-      "thrust_level",
-      "solid_propellant_mass",
-      "inclination",
-      "depth",
+      "height_of_pericentre",
+      "comment",
+      "reentry",
+      "latitude",
+      "cospar_launch_number",
+      "height",
       "max_cross_section",
-      "wet_mass",
-      "span",
-      "oxidiser",
-      "eccentricity",
-      "burn_time",
       "height_of_centre",
-      "gto_capacity",
-      "mean_anomaly",
-      "constraint",
+      "width",
       "fuel_mass",
-      "geo_capacity",
+      "depth",
       "sso_capacity",
-      "fuel",
+      "oxidiser_mass",
+      "arg_of_periapsis",
+      "gto_capacity",
       "vimpel_id",
+      "failed_launches",
+      "esc_capacity",
+      "dry_mass",
       "azimuths",
+      "max_specific_impulse",
+      "oxidiser",
       "leo_capacity",
-      "oxidiser_mass",
-      "min_cross_section",
-      "height_of_pericentre",
-      "latitude",
-      "successful_launches",
-      "comment",
+      "geo_capacity",
+      "eccentricity",
       "fragmentation_event_type",
-      "dry_mass",
-      "satno",
-      "arg_of_periapsis",
-      "longitude",
+      "failure",
+      "span",
+      "thrust_level",
+      "mean_anomaly",
+      "solid_propellant",
       "discos_id",
-      "raan",
-      "number_of_stages",
+      "constraint",
+      "inclination",
+      "satno",
       "cospar_id",
-      "solid_propellant",
-      "reentry",
-      "mass",
-      "width",
-      "max_specific_impulse",
-      "cospar_launch_number",
-      "shape",
-      "altitude"
+      "shape"
     ],
     "id2concept": [
       "<PAD>",
@@ -182507,5 +182494,5 @@
       "state": 2
     }
   },
-  "vocab_size": 28996
+  "vocab_size": 50265
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:927a6cf76a40ee59148de5e3d8f89a286fa095a2303527045fa68a941973ee27
-size 479274271
+oid sha256:cda6fb7cd856ea13f9f94d53524ac55a536c7a7e6bc016f9616084d688f9e6e6
+size 544616351