Relphormer baseline

pretrain/models/model.py

@@ -0,0 +1,10 @@
+from transformers.models.bert.modeling_bert import BertForMaskedLM
+
+
+
+
+class BertKGC(BertForMaskedLM):
+    @staticmethod
+    def add_to_argparse(parser):
+        parser.add_argument("--pretrain", type=int, default=0, help="")
+        return parser