Create config.json

config.json

@@ -0,0 +1,67 @@
+{
+  "activation_dropout": 0.1,
+  "activation_fn": "gelu",
+  "apply_graphormer_init": true,
+  "architectures": [
+    "GraphormerForGraphClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bias": true,
+  "bos_token_id": 1,
+  "dropout": 0.0,
+  "edge_type": "multi_hop",
+  "embed_scale": null,
+  "embedding_dim": 768,
+  "encoder_normalize_before": true,
+  "eos_token_id": 2,
+  "export": false,
+  "ffn_embedding_dim": 768,
+  "freeze_embeddings": false,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "a",
+    "1": "b",
+    "2": "c",
+    "3": "d",
+    "4": "e",
+    "5": "f",
+    "6": "g"
+  },
+  "init_fn": null,
+  "kdim": null,
+  "label2id": {
+    "a": 0,
+    "b": 1,
+    "c": 2,
+    "d": 3,
+    "e": 4,
+    "f": 5,
+    "g": 6
+  },
+  "layerdrop": 0.0,
+  "max_nodes": 512,
+  "model_type": "graphormer",
+  "multi_hop_max_dist": 5,
+  "no_token_positional_embeddings": false,
+  "num_atoms": 4608,
+  "num_attention_heads": 32,
+  "num_edge_dis": 128,
+  "num_edges": 1536,
+  "num_in_degree": 512,
+  "num_layers": 12,
+  "num_out_degree": 512,
+  "num_spatial": 512,
+  "num_trans_layers_to_freeze": 0,
+  "num_classes": 7,           // Add the number of classes
+  "pad_token_id": 0,
+  "pre_layernorm": false,
+  "q_noise": 0.0,
+  "qn_block_size": 8,
+  "self_attention": true,
+  "share_input_output_embed": false,
+  "spatial_pos_max": 1024,
+  "torch_dtype": "float32",
+  "traceable": false,
+  "transformers_version": "4.26.0.dev0",
+  "vdim": null
+}